Атрибут ndim
Атрибут ndim класса Tensor возвращает целое число,
равное количеству измерений (осей) тензора.
Атрибут доступен только для чтения и не принимает параметров.
Для скалярного тензора значение равно 0,
для вектора - 1, для матрицы - 2 и так далее.
Атрибут ndim эквивалентен длине кортежа shape.
Синтаксис
t.ndim
Пример
Давайте создадим скалярный тензор и выведем число его измерений:
import tensorflow as tf
t = tf.constant(5)
print(t.ndim)
Результат выполнения кода:
0
Пример
Давайте создадим вектор и выведем число его измерений:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
print(t.ndim)
Результат выполнения кода:
1
Пример
Давайте создадим матрицу и выведем число ее измерений:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
print(t.ndim)
Результат выполнения кода:
2
Пример
Давайте сравним атрибут ndim с длиной кортежа shape:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
print(t.ndim)
print(len(t.shape))
Результат выполнения кода:
2
2
