Функция size
Функция size возвращает общее количество элементов в тензоре.
Первым параметром функция принимает тензор или массив.
Вторым параметром можно передать имя для операции.
Функция полезна для динамического определения размера данных,
например, при работе с изменяемыми тензорами или в графовом режиме.
Синтаксис
tf.size(input, [name])
Пример
Давайте создадим одномерный тензор и получим количество его элементов:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
res = tf.size(t)
print(res)
Результат выполнения кода:
tf.Tensor(5, shape=(), dtype=int32)
Пример
Давайте создадим двумерный тензор и получим количество его элементов:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.size(t)
print(res)
Результат выполнения кода:
tf.Tensor(6, shape=(), dtype=int32)
Пример
Давайте получим значение размера в виде числа NumPy:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.size(t).numpy()
print(res)
Результат выполнения кода:
6
