tf.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.cast(x, tf.int32)
<tf.Tensor: shape=(2,), dtype=int32, numpy=array([1, 2], dtype=int32)>

Обратите внимание, что tf.cast имеет псевдоним tf.dtypes.cast:

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 значений в целочисленные типы не определено.

Обратите внимание, что эта операция может привести к потере точности при преобразовании нативных Python-переменных float и complex в tf.float64 или tf.complex128 тензоры, поскольку вход сначала преобразуется в тип float32, а затем расширяется. Рекомендуется использовать tf.convert_to_tensor вместо tf.cast для любых входных данных, которые не являются тензорами.