tf.compat.v1.to_bfloat16

Преобразует тензор к типу bfloat16. (устарело)

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

Перейдите на TF2

Это имя устарело и было удалено в TF2, но ему на смену пришла точная замена tf.cast(..., tf.bfloat16). При выполнении с eager execution или tf.function дополнительных проблем не возникает.

До:

tf.compat.v1.to_bfloat16(tf.constant(3.14, dtype=tf.float32))
<tf.Tensor: shape=(), dtype=bfloat16, numpy=3.14>

После:

tf.cast(tf.constant(3.14, dtype=tf.float32), tf.bfloat16)
<tf.Tensor: shape=(), dtype=bfloat16, numpy=3.14>

Описание

Аргументы
`x`	`Tensor`, `SparseTensor` или `IndexedSlices`.
`name`	Имя операции (необязательно).

Возвращает
`Tensor`, `SparseTensor` или `IndexedSlices` той же формы, что и `x`, с типом `bfloat16`.

Вызывает исключения
`TypeError`	Если `x` нельзя преобразовать к `bfloat16`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/to_bfloat16