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/api_docs/python/tf/compat/v1/to_bfloat16
