tf.compat.v1.to_bfloat16

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

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

Миграция в TF2

Это имя устарело и удалено в TF2, но имеет точный заменитель tf.cast(..., tf.bfloat16). Нет дополнительных проблем с выполнением eager или 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`	A `Tensor` или `SparseTensor` или `IndexedSlices`.
`name`	Имя операции (необязательно).

Возвращаемое значение
A `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/versions/r2.9/api_docs/python/tf/compat/v1/to_bfloat16