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


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)

返回

创建的数组

返回类型

ND阵列

返回一个二维数组,其中对角线为 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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。