當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。