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


Python mxnet.ndarray.array用法及代码示例


用法:

mxnet.ndarray.array(source_array, ctx=None, dtype=None)

参数

  • source_array(array_like) - 一个暴露数组接口的对象,一个对象__array__方法返回一个数组或任何(嵌套)序列。
  • ctx(mxnet.context.Context, optional) - 设备上下文(默认为当前默认上下文)。
  • dtype(str or numpy.dtype, optional) - 输出数组的数据类型。默认数据类型是source_array.dtype如果source_array是一个NDArray,float32否则。

返回

source_array 内容相同的数组。

返回类型

NDArrayRowSparseNDArrayCSRNDArray

从任何公开数组接口的对象创建一个数组。

例子

>>> import numpy as np
>>> mx.nd.array([1, 2, 3])
<NDArray 3 @cpu(0)>
>>> mx.nd.array([[1, 2], [3, 4]])
<NDArray 2x2 @cpu(0)>
>>> mx.nd.array(np.zeros((3, 2)))
<NDArray 3x2 @cpu(0)>
>>> mx.nd.array(np.zeros((3, 2)), mx.gpu(0))
<NDArray 3x2 @gpu(0)>
>>> mx.nd.array(mx.nd.zeros((3, 2), stype='row_sparse'))
<RowSparseNDArray 3x2 @cpu(0)>

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.array。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。