tf.raw_ops.Reshape
Изменяет форму тензора.
tf.raw_ops.Reshape(
tensor, shape, name=None
)
Учитывая tensor, эта операция возвращает тензор, имеющий те же значения, что и tensor, с формой shape.
Если один компонент 1-мерного тензора shape имеет специальное значение -1, размер этого измерения вычисляется таким образом, чтобы общий размер оставался постоянным. В частности, shape размера [-1] сглаживается в 1-мерный. Не более одного компонента shape может быть неизвестным.
shape должен быть 1-мерным, и операция возвращает тензор с формой shape, заполненной значениями tensor. В этом случае количество элементов, подразумеваемых shape должно совпадать с количеством элементов в tensor.
Ошибка возникает, если shape не является 1-мерным.
Например:
# tensor 't' is [1, 2, 3, 4, 5, 6, 7, 8, 9]
# tensor 't' has shape [9]
reshape(t, [3, 3]) ==> [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
# tensor 't' is [[[1, 1], [2, 2]],
# [[3, 3], [4, 4]]]
# tensor 't' has shape [2, 2, 2]
reshape(t, [2, 4]) ==> [[1, 1, 2, 2],
[3, 3, 4, 4]]
# tensor 't' is [[[1, 1, 1],
# [2, 2, 2]],
# [[3, 3, 3],
# [4, 4, 4]],
# [[5, 5, 5],
# [6, 6, 6]]]
# tensor 't' has shape [3, 2, 3]
# pass '[-1]' to flatten 't'
reshape(t, [-1]) ==> [1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6]
# -1 can also be used to infer the shape
# -1 is inferred to be 9:
reshape(t, [2, -1]) ==> [[1, 1, 1, 2, 2, 2, 3, 3, 3],
[4, 4, 4, 5, 5, 5, 6, 6, 6]]
# -1 is inferred to be 2:
reshape(t, [-1, 9]) ==> [[1, 1, 1, 2, 2, 2, 3, 3, 3],
[4, 4, 4, 5, 5, 5, 6, 6, 6]]
# -1 is inferred to be 3:
reshape(t, [ 2, -1, 3]) ==> [[[1, 1, 1],
[2, 2, 2],
[3, 3, 3]],
[[4, 4, 4],
[5, 5, 5],
[6, 6, 6]]]
# tensor 't' is [7]
# shape `[]` reshapes to a scalar
reshape(t, []) ==> 7
| Аргументы | |
|---|---|
tensor |
А Tensor |
shape |
А Tensor. Должен быть одним из следующих типов: int32, int64. Определяет форму выходного тензора. |
name |
Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
А Tensor. Имеет тот же тип, что и tensor |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/Reshape