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

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

用法:
mxnet.ndarray.eye(N, M=0, k=0, ctx=None, dtype=None, **kwargs)

參數：

N：(int) - 輸出中的行數。
M：(int, optional) - 輸出中的列數。如果為 0，則默認為 N。
k：(int, optional) - 對角線索引：0(默認)表示主對角線，正值表示上對角線，負值表示下對角線。
ctx：(mxnet.context.Context, optional) - 一個可選的設備上下文(默認是當前的默認上下文)
dtype：(str or numpy.dtype, optional) - 可選值類型(默認為float32)

返回：

創建的數組

返回類型：

返回一個二維數組，其中對角線為 1，其他位置為 0。

例子：

>>> mx.nd.eye(2)
[[ 1.  0.]
 [ 0.  1.]]
<NDArray 2x2 @cpu(0)>
>>> mx.nd.eye(2, 3, 1)
[[ 0.  1.  0.]
 [ 0.  0.  1.]]
<NDArray 2x3 @cpu(0)>

相關用法

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