cupy.where#

cupy.where(condition, x=None, y=None)[source]#

根据 condition 返回 x 或 y 中的元素。

如果只给定 condition，则返回 condition.nonzero()。

参数:

  • condition (cupy.ndarray) – 当为 True 时，选择 x，否则选择 y。

  • x (cupy.ndarray) – 当为 True 时从中选择的值。

  • y (cupy.ndarray) – 当为 False 时从中选择的值。

返回:

当 condition 为 True 时，输出的每个元素包含 x 中的元素，否则包含 y 中的元素。如果只给定 condition，则返回元组 condition.nonzero()，即 condition 为 True 的索引。

返回类型:

cupy.ndarray

警告

如果同时省略了 x 和 y，则此函数可能会同步设备。

另请参阅

numpy.where()