tf.raw_ops.ConcatOffset
Вычисляет смещения входных данных операции конкатенации в выходном тензоре.
tf.raw_ops.ConcatOffset(
concat_dim, shape, name=None
)
Например:
x = [2, 2, 7] y = [2, 3, 7] z = [2, 9, 7] offsets = concat_offset(1, [x, y, z]) [list(off.numpy()) for off in offsets] [[0, 0, 0], [0, 2, 0], [0, 5, 0]]
Обычно это используется при вычислении градиентов для операции конкатенации.
| Аргументы | |
|---|---|
concat_dim | Tensor типа int32. Измерение, вдоль которого выполняется конкатенация. |
shape | Список как минимум из 2 объектов Tensor с одинаковым типом из: int32, int64. N — векторы int32 или int64, представляющие форму конкатенируемых тензоров. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Список той же длины, что и shape, состоящий из объектов Tensor с тем же типом, что и shape. |
© 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/ConcatOffset