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