TensorFlow函數(shù)：tf.to_bfloat16

tf.to_bfloat16函數(shù)

tf.to_bfloat16(
    x,
    name='ToBFloat16'
)

定義在：tensorflow/python/ops/math_ops.py.

參見指南：張量轉(zhuǎn)換

將張量轉(zhuǎn)換為 bfloat16 類型.

函數(shù)參數(shù)：

• x：一個 Tensor 或 SparseTensor.
• name：操作的名稱(可選).

函數(shù)返回值：

tf.to_bfloat16函數(shù)返回一個 Tensor 或 SparseTensor 與 x(類型為 bfloat16)具有相同的形狀.

可能引發(fā)的異常：

• TypeError：如果 x 不能轉(zhuǎn)換為 bfloat16.