tf.raw_ops.Squeeze
Удаляет размерность размером 1 из формы тензора.
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