Функция shape

Функция shape возвращает форму тензора в виде объекта TensorShape. Первым параметром функция принимает тензор. Вторым параметром можно передать тип данных out_type для элементов формы. Форма показывает размерность тензора по каждому измерению.

Синтаксис

tf.shape(input, [out_type])

Пример

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

import tensorflow as tf

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

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

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

Пример

Давайте создадим двумерный тензор и получим его форму:

import tensorflow as tf

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

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

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

Пример

Давайте используем атрибут shape тензора для получения формы:

import tensorflow as tf

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

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

(2, 3)

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

функцию rank,
которая возвращает ранг тензора
функцию size,
которая возвращает количество элементов в тензоре
функцию reshape,
которая изменяет форму тензора
функцию ensure_shape,
которая проверяет и устанавливает форму тензора