Функция convert_to_tensor

Функция convert_to_tensor преобразует переданные данные в тензор TensorFlow. Первым параметром функция принимает число, список, массив NumPy или другой тензор. Вторым параметром можно передать тип данных элементов.

В отличие от функции constant, которая всегда создает новый тензор, функция convert_to_tensor возвращает уже существующий тензор без копирования, если переданный объект им является. Это делает функцию удобной для приведения разнородных данных к единому типу.

Синтаксис

tf.convert_to_tensor(value, [dtype])

Пример

Давайте преобразуем список чисел 1, 2, 3, 4, 5 в тензор:

import tensorflow as tf

t = tf.convert_to_tensor([1, 2, 3, 4, 5])
print(t)

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

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

Пример

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

import tensorflow as tf

t = tf.convert_to_tensor([1, 2, 3, 4, 5], dtype=tf.float32)
print(t)

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

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

Пример

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

import tensorflow as tf

t = tf.convert_to_tensor([[1, 2, 3], [4, 5, 6]])
print(t)

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

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

Пример

Давайте преобразуем строку в тензор:

import tensorflow as tf

t = tf.convert_to_tensor('abcde')
print(t)

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

tf.Tensor(b'abcde', shape=(), dtype=string)

Пример

Давайте преобразуем массив NumPy в тензор:

import tensorflow as tf
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
t = tf.convert_to_tensor(arr)
print(t)

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

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

Пример

Давайте передадим в функцию уже существующий тензор и убедимся, что он возвращается без изменений:

import tensorflow as tf

t1 = tf.constant([1, 2, 3, 4, 5])
t2 = tf.convert_to_tensor(t1)
print(t1 is t2)

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

True

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

функцию constant,
которая создает тензор из переданных данных
функцию cast,
которая приводит тензор к заданному типу данных
функцию make_ndarray,
которая преобразует тензор в массив NumPy
функцию shape,
которая возвращает форму тензора