Как использовать num_elements из TensorFlow?

Question

Мне просто нужно количество элементов в моем тензорном, независимо от формы.

Я вижу в документации num_elements служит моей цели. Однако, если я пытаюсь использовать его как myTensor.num_elements(), я получаю эту ошибку:

AttributeError: у объекта 'Tensor' нет атрибута 'num_elements'

Как сделатьЯ делаю это?

Answer 1

Этот метод относится к TensorShape вместо Tensor. Вы можете получить количество элементов, подобных этому

import tensorflow as tf
a = tf.random.uniform((1,16,23))
a.shape.num_elements()
# 368