cunumeric.putmask

cunumeric.putmask(a, mask, values)

Changes elements of an array based on conditional and input values. Sets a.flat[n] = values[n] for each n where mask.flat[n]==True. If values is not the same size as a and mask then it will repeat. This gives behavior different from a[mask] = values.

Parameters:

• a (ndarray) – Target array.

• mask (array_like) – Boolean mask array. It has to be the same shape as a.

• values (array_like) – Values to put into a where mask is True. If values is smaller than a it will be repeated.

See also

numpy.putmask

Availability:

Multiple GPUs, Multiple CPUs