Метод get_shape
Метод get_shape класса Tensor возвращает форму тензора в виде объекта TensorShape. Метод не принимает параметров и применяется к объекту тензора. Полученная форма содержит размерности тензора по каждому измерению. Для статических тензоров форма известна на момент построения графа вычислений.
Синтаксис
t.get_shape()
Пример
Давайте создадим одномерный тензор и получим его форму:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
res = t.get_shape()
print(res)
Результат выполнения кода:
(5,)
Пример
Давайте создадим двумерный тензор и получим его форму:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = t.get_shape()
print(res)
Результат выполнения кода:
(2, 3)
Пример
Давайте получим отдельные размерности тензора через индексацию формы:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = t.get_shape()
print(res[0])
print(res[1])
Результат выполнения кода:
2
3
Пример
Давайте сравним результат метода get_shape и атрибута shape:
<+python+>
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
res = t.get_shape()
print(res)
print(t.shape)
<-python+>
Результат выполнения кода:
(5,)
(5,)
