Spec-Zone.ru › TensorFlow

tf.size

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

tf.size(
    input, out_type=None, name=None
)

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

Используется в руководстве Используется в учебных пособиях
  • Аппроксимация матриц с помощью Core API
  • Введение в тензоры
  • Предварительная обработка BERT с помощью TF Text
  • Масштабируемое сжатие моделей
  • Эффективное с точки зрения клиента федеративное обучение больших моделей с помощью `federated_select` и разреженной агрегации
  • Отправка разных данных определенным клиентам с помощью tff.federated_select
  • Классификация поз человека с помощью MoveNet и TensorFlow Lite
  • Федеративное обучение для классификации изображений

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

Возвращает скаляр Tensor, представляющий количество элементов в input типа out_type. По умолчанию — tf.int32.

Например:

t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.size(t)
<tf.Tensor: shape=(), dtype=int32, numpy=12>
Аргументы
input Объект типа Tensor или SparseTensor.
out_type (Необязательно) Указанный числовой тип выходных данных операции без квантования. По умолчанию — tf.int32. (Примечание: существует экспериментальный флаг tf_shape_default_int64, который меняет значение по умолчанию на tf.int64. Это неподдерживаемая экспериментальная настройка, вызывающая известные сбои.)
name Имя операции (необязательно).
Возвращает
Объект типа Tensor и out_type. По умолчанию — tf.int32.

Совместимость с NumPy

Эквивалентно np.size()

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/size

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API