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


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。