tf.raw_ops.Pack
Упаковывает список N тензоров ранга R в один тензор ранга (R+1).
tf.raw_ops.Pack(
values, axis=0, name=None
)
Упаковывает тензоры N в values в тензор с рангом на единицу выше, чем у каждого тензора в values, упаковывая их вдоль axis измерения. При заданном списке тензоров формы (A, B, C);
если axis == 0, то тензор output будет иметь форму (N, A, B, C). если axis == 1, то тензор output будет иметь форму (A, N, B, C). И так далее.
Пример:
# 'x' is [1, 4] # 'y' is [2, 5] # 'z' is [3, 6] pack([x, y, z]) => [[1, 4], [2, 5], [3, 6]] # Pack along first dim. pack([x, y, z], axis=1) => [[1, 2, 3], [4, 5, 6]]
Это противоположно unpack.
| Аргументы | |
|---|---|
values | Список из по меньшей мере 1 Tensor объекта с одинаковым типом. Должны иметь одинаковую форму и тип. |
axis | Необязательный int. По умолчанию равен 0. Измерение, по которому происходит упаковка. Отрицательные значения оборачиваются, поэтому допустимый диапазон — [-(R+1), R+1). |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor. Имеет тот же тип, что и values. |
© 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/Pack