cupy.place#

cupy.place(arr, mask, vals)[source]#

根据条件和输入值改变数组的元素。

此函数使用 vals 中的前 N 个元素，其中 N 是 mask 中为 True 的值的数量。

参数:

  • arr (cupy.ndarray) – 要放入数据的数组。

  • mask (array-like) – 布尔掩码数组。必须与 a 的大小相同。

  • vals (array-like) – 要放入 a 中的值。仅使用前 N 个元素，其中 N 是 mask 中为 True 的值的数量。如果 vals 小于 N，它将被重复使用；如果要掩码 a 中的元素，则此序列不能为空。

示例

>>> arr = np.arange(6).reshape(2, 3)
>>> np.place(arr, arr>2, [44, 55])
>>> arr
array([[ 0,  1,  2],
       [44, 55, 44]])

警告

此函数可能会同步设备。

另请参阅

numpy.place()