Функция cast

Функция cast преобразует тензор в заданный тип данных. Первым параметром функция принимает тензор, который нужно преобразовать. Вторым параметром передается целевой тип данных dtype. Третьим необязательным параметром можно передать имя операции. Функция возвращает новый тензор с тем же содержимым, но с новым типом элементов.

Синтаксис

tf.cast(x, dtype, [name])

Пример

Давайте преобразуем тензор целых чисел в тип float32:

import tensorflow as tf

t = tf.constant([1, 2, 3, 4, 5])
res = tf.cast(t, tf.float32)
print(res)

Результат выполнения кода:

tf.Tensor([1. 2. 3. 4. 5.], shape=(5,), dtype=float32)

Пример

Давайте преобразуем тензор дробных чисел в целочисленный тип int32:

<+python+> import tensorflow as tf t = tf.constant([1.7, 2.3, 3.9, 4.1, 5.5]) res = tf.cast(t, tf.int32) print(res) <-python+>

Результат выполнения кода:

tf.Tensor([1 2 3 4 5], shape=(5,), dtype=int32)

Пример

Давайте преобразуем двумерный тензор в тип float64:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.cast(t, tf.float64)
print(res)

Результат выполнения кода:

tf.Tensor(
[[1. 2. 3.]
 [4. 5. 6.]], shape=(2, 3), dtype=float64)

Пример

Давайте преобразуем целочисленный тензор в логический тип bool:

import tensorflow as tf

t = tf.constant([0, 1, 2, 0, 3])
res = tf.cast(t, tf.bool)
print(res)

Результат выполнения кода:

tf.Tensor([False True True False True], shape=(5,), dtype=bool)

Смотрите также

функцию convert_to_tensor,
которая преобразует данные в тензор
функцию constant,
которая создает тензор из переданных данных
функцию TensorSpec,
которая описывает форму и тип тензора
функцию reshape,
которая изменяет форму тензора