tf.rank

Возвращает ранг тензора.

tf.rank(
    input, name=None
)

Используется в блокнотах

См. также tf.shape.

Возвращает 0-D int32 Tensor, представляющее ранг input.

Например:

# shape of tensor 't' is [2, 2, 3]
t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.rank(t)  # 3

Примечание: Ранг тензора — это не то же самое, что ранг матрицы. Ранг тензора — это число индексов, необходимых для однозначного выбора каждого элемента тензора. Ранг также называют «порядком», «степенью» или «ndims».

Эквивалентно np.ndim