当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.ndarray.where用法及代码示例


用法:

mxnet.ndarray.where(condition=None, x=None, y=None, out=None, name=None, **kwargs)

参数：

返回：

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

根据条件从 x 或 y 返回元素。

给定三个 ndarray，condition、x 和 y，返回一个 ndarray，其中包含 x 或 y 中的元素，具体取决于 condition 中的元素是 true 还是 false。 x 和 y 必须具有相同的形状。如果条件与 x 具有相同的形状，则如果条件中的相应元素为真，则输出数组中的每个元素都来自 x，如果条件中的相应元素为假，则来自 y。

如果条件与 x 的形状不同，则它必须是大小与 x 的第一维大小相同的一维数组。如果条件中的对应元素为真，则输出数组的每一行都来自 x 的行，如果为假，则来自 y 的行。

请注意，所有非零值都被解释为条件中的True。

例子：

x = [[1, 2], [3, 4]]
y = [[5, 6], [7, 8]]
cond = [[0, 1], [-1, 0]]

where(cond, x, y) = [[5, 2], [3, 8]]

csr_cond = cast_storage(cond, 'csr')

where(csr_cond, x, y) = [[5, 2], [3, 8]]

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.where。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。