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

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