tf.cast

Преобразует тензор в новый тип.

Просмотр псевдонимов

Основные псевдонимы

tf.dtypes.cast

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.cast, tf.compat.v1.dtypes.cast

tf.cast(
    x, dtype, name=None
)

Операция преобразует x (в случае Tensor) или x.values (в случае SparseTensor или IndexedSlices) в dtype.

Например:

x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.dtypes.cast(x, tf.int32)
<tf.Tensor: shape=(2,), dtype=int32, numpy=array([1, 2], dtype=int32)>

Операция поддерживает типы данных (для x и dtype) uint8, uint16, uint32, uint64, int8, int16, int32, int64, float16, float32, float64, complex64, complex128, bfloat16. В случае преобразования из комплексных типов (complex64, complex128) в вещественные типы возвращается только вещественная часть x. В случае преобразования из вещественных типов в комплексные типы (complex64, complex128) мнимая часть возвращаемого значения устанавливается в 0. Обработка комплексных типов здесь соответствует поведению numpy.

Обратите внимание, что преобразование значений nan и inf в целочисленные типы имеет неопределённое поведение.

Аргументы
`x`	`Tensor` или `SparseTensor` или `IndexedSlices` числового типа. Может быть `uint8`, `uint16`, `uint32`, `uint64`, `int8`, `int16`, `int32`, `int64`, `float16`, `float32`, `float64`, `complex64`, `complex128`, `bfloat16` .
`dtype`	Тип назначения. Список поддерживаемых типов данных такой же, как у `x`.
`name`	Имя операции (необязательно).

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

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

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/cast