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 将在输入可用时将其复制到输出中，在某些情况下，这可以提供性能优势。