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/versions/r2.9/api_docs/python/tf/raw_ops/Pack