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


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。