Python mxnet.ndarray.equal用法及代碼示例

用法: mxnet.ndarray.equal(lhs, rhs)

參數：

lhs：(scalar or mxnet.ndarray.array) - 要比較的第一個數組。
rhs：(scalar or mxnet.ndarray.array) - 要比較的第二個數組。如果lhs.shape != rhs.shape，它們必須可以廣播到一個共同的形狀。

布爾值的輸出數組。

返回類型：

ND陣列

返回元素方式的結果等於(==) 與廣播的比較操作。

對於輸入數組中的每個元素，如果對應的元素相同，則返回 1(true)，否則返回 0(false)。

等效於 lhs == rhs 和 mx.nd.broadcast_equal(lhs, rhs) 。

注意：

如果兩個數組的對應維度具有相同的大小或其中一個的大小為 1，則這些數組可以廣播為一個共同的形狀。

例子：

>>> x = mx.nd.ones((2,3))
>>> y = mx.nd.arange(2).reshape((2,1))
>>> z = mx.nd.arange(2).reshape((1,2))
>>> x.asnumpy()
array([[ 1.,  1.,  1.],
       [ 1.,  1.,  1.]], dtype=float32)
>>> y.asnumpy()
array([[ 0.],
       [ 1.]], dtype=float32)
>>> z.asnumpy()
array([[ 0.,  1.]], dtype=float32)
>>> (x == 1).asnumpy()
array([[ 1.,  1.,  1.],
       [ 1.,  1.,  1.]], dtype=float32)
>>> (x == y).asnumpy()
array([[ 0.,  0.,  0.],
       [ 1.,  1.,  1.]], dtype=float32)
>>> mx.nd.equal(x,y).asnumpy()
array([[ 0.,  0.,  0.],
       [ 1.,  1.,  1.]], dtype=float32)
>>> (z == y).asnumpy()
array([[ 1.,  0.],
       [ 0.,  1.]], dtype=float32)

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.equal。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：

注意：

例子：