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`	A `Tensor`. Сжимаемое `input`.
`axis`	Необязательный список `ints`. По умолчанию `[]`. Если указан, сжимаются только перечисленные измерения. Индекс измерения начинается с 0. Ошибка, если сжимается измерение, не равное 1. Должен быть в диапазоне `[-rank(input), rank(input))`. Должен быть указан, если `input` является `RaggedTensor`.
`name`	Имя операции (необязательно).
`squeeze_dims`	Устаревший ключевой аргумент, теперь axis.

Возвращаемое значение
A `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