Python tf.raw_ops.ParallelConcat用法及代碼示例

沿第一維連接 N 張量列表。

用法

tf.raw_ops.ParallelConcat(
    values, shape, name=None
)

輸入張量都必須在第一維中具有大小 1。

例如：

# 'x' is [[1, 4]]
# 'y' is [[2, 5]]
# 'z' is [[3, 6]]
parallel_concat([x, y, z]) => [[1, 4], [2, 5], [3, 6]]  # Pack along first dim.

concat 和 parallel_concat 之間的區別在於 concat 要求在操作開始之前計算所有輸入，但不要求在圖形構建期間知道輸入形狀。並行 concat 將在輸入可用時將其複製到輸出中，在某些情況下，這可以提供性能優勢。