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

用法:
mxnet.symbol.where(condition=None, x=None, y=None, name=None, attr=None, out=None, **kwargs)

参数：

condition：(Symbol) - 条件数组
x：(Symbol) -
y：(Symbol) -
name：(string, optional.) - 结果符号的名称。

结果符号。

返回类型：

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

用法:

参数：

返回：

返回类型：