tf.raw_ops.AssignVariableXlaConcatND
Конкатенирует входной тензор по всем измерениям.
tf.raw_ops.AssignVariableXlaConcatND(
resource, inputs, num_concats, paddings=[], name=None
)
Операция, которая объединяет фрагменты входного тензора, основанные на атрибуте num_splits, необязательно удаляет заполнители и записывает объединённый тензор без заполнителей в ресурсный переменную.
Эта операция может быть сгенерирована через мост TPU.
Например, для тензора input:
[[0, 1], [4, 5]] [[2, 3], [6, 7]] [[8, 9], [12, 13]] [[10, 11], [14, 15]]
num_splits:
[2, 2]
и paddings:
[1, 1]
ожидаемый outputs результат:
[[0, 1, 2], [4, 5, 6], [8, 9, 10]]
| Аргументы | |
|---|---|
resource | A Tensor of type resource. Ресурсная переменная для конкатенированных входных тензоров по всем измерениям. } in_arg { name: "inputs" description: <
|
inputs | Список из, как минимум, 1 Tensor объектов с одинаковым типом. |
num_concats | Список ints. Количество способов объединения по измерениям. |
paddings | Необязательный список ints. По умолчанию []. Необязательный список правых заполнителей по измерениям, которые нужно удалить из конечного объединённого тензора. Эти заполнители не должны превышать размер измерения объединённого результата до удаления заполнителей. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 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/AssignVariableXlaConcatND
