Spec-Zone.ru › TensorFlow

tf.raw_ops.Squeeze

Удаляет размерность размером 1 из формы тензора.

Псевдонимы

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.Squeeze

tf.raw_ops.Squeeze(
    input, axis=[], name=None
)

Для данного тензора input, эта операция возвращает тензор того же типа со всеми размерностями размера 1, удалёнными. Если вы не хотите удалять все размерности размера 1, вы можете удалить определённые размерности размера 1, указав axis.

Например:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t)) ==> [2, 3]

Или, чтобы удалить определённые размерности размера 1:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t, [2, 4])) ==> [1, 2, 3, 1]
Аргументы
input A Tensor. Тензор для сжатия.
axis Необязательный список ints. По умолчанию []. Если указано, то сжимаются только указанные размерности. Индекс размерности начинается с 0. Ошибка, если сжимается размерность, не равная 1. Должен быть в диапазоне [-rank(input), rank(input)).
name Имя операции (необязательно).
Возвращаемое значение
A Tensor. Имеет тот же тип, что и input.

© 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/raw_ops/Squeeze

Spec-Zone.ru

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