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

Python mxnet.ndarray.NDArray.astype用法及代碼示例


用法:

astype(dtype, copy=True)

參數：

  • dtype：(numpy.dtype or str) - 返回數組的類型。
  • copy：(bool) - 默認True.默認情況下， astype 總是在相同的上下文中返回一個新分配的 ndarray。如果設置為False，並且請求的 dtype 與 ndarray 的 dtype 相同，則返回 ndarray 而不是副本。

返回：

轉換為指定類型後複製的數組，如果 copy=False 且 dtype 與輸入數組相同，則為相同的數組。

返回類型：

NDArray、CSRNDArray或RowSparseNDArray

轉換為指定類型後返回數組的副本。

例子：

>>> x = mx.nd.zeros((2,3), dtype='float32')
>>> y = x.astype('int32')
>>> y.dtype
<type 'numpy.int32'>

相關用法


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