Число элементов в TensorFlow

Число элементов - это сколько всего значений лежит в тензоре. Форма описывает раскладку по осям, а общий счёт даёт функция size.

У одномерного тензора элементов ровно столько, какова длина его единственной оси. Создадим ряд из четырёх чисел и выведем общее число элементов:

import tensorflow as tf

scores = tf.constant([1, 2, 3, 4])
print(tf.size(scores))

Если в тензоре один элемент, функция size вернёт 1:

import tensorflow as tf

prices = tf.constant([99])
print(tf.size(prices))

У таблицы считают все ячейки сразу, независимо от числа строк и столбцов:

import tensorflow as tf

grid = tf.constant([[10, 20], [30, 40]])
print(tf.size(grid))

Соберите тензор из чисел 11, 22 и 33 и выведите, сколько в нём значений всего.

Постройте таблицу [[5, 6], [7, 8]] и выведите общее число её ячеек.

Соберите тензор из шести целых чисел подряд и выведите, сколько в нём элементов.