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. Тензор input, размерность которого нужно сжать. |
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