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

Python mxnet.ndarray.op.cast用法及代碼示例

用法:

mxnet.ndarray.op.cast(data=None, dtype=_Null, out=None, name=None, **kwargs)

參數：

  • data：(NDArray) - 輸入。
  • dtype：({'bfloat16', 'bool', 'float16', 'float32', 'float64', 'int32', 'int64', 'int8', 'uint8'}, required) - 輸出數據類型。
  • out：(NDArray, optional) - 輸出 NDArray 來保存結果。

返回：

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

將輸入的所有元素轉換為新類型。

注意：

Cast 已棄用。請改用cast。

例子：

cast([0.9, 1.3], dtype='int32') = [0, 1]
cast([1e20, 11.1], dtype='float16') = [inf, 11.09375]
cast([300, 11.1, 10.9, -1, -3], dtype='uint8') = [44, 11, 10, 255, 253]

相關用法


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