tf.compat.v1.squeeze

Удаляет измерения размера 1 из формы тензора. (устаревшие аргументы)

tf.compat.v1.squeeze(
    input, axis=None, name=None, squeeze_dims=None
)

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

Используется в учебных материалах
Языковые модели Wiki40B

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

Например:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
t = tf.ones([1, 2, 1, 3, 1, 1])
print(tf.shape(tf.squeeze(t)).numpy())
[2 3]

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

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
t = tf.ones([1, 2, 1, 3, 1, 1])
print(tf.shape(tf.squeeze(t, [2, 4])).numpy())
[1 2 3 1]

Примечание: если input является tf.RaggedTensor, эта операция выполняется за O(N) времени, где N — количество элементов в сжимаемых измерениях.

Аргументы
`input`	`Tensor`. `input`, который нужно сжать.
`axis`	Необязательный список `ints`. По умолчанию — `[]`. Если указано, сжимаются только перечисленные измерения. Индексация измерений начинается с 0. Ошибкой является попытка сжать измерение, размер которого не равен 1. Значение должно находиться в диапазоне `[-rank(input), rank(input))`. Необходимо указать, если `input` — это `RaggedTensor`.
`name`	Имя операции (необязательно).
`squeeze_dims`	Устаревший именованный аргумент, который теперь называется axis.

Возвращает
`Tensor`. Имеет тот же тип, что и `input`. Содержит те же данные, что и `input`, но с одним или несколькими удалёнными измерениями размера 1.

Вызывает исключение
`ValueError`	Если указаны одновременно `squeeze_dims` и `axis`.

© 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/compat/v1/squeeze