当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.raw_ops.TPUReplicatedOutput用法及代码示例


连接来自 N-way 复制 TPU 计算的 N 个输出。

用法

tf.raw_ops.TPUReplicatedOutput(
    input, num_replicas, name=None
)

参数

  • input 一个Tensor。
  • num_replicas int 即 >= 1 。
  • name 操作的名称(可选)。

返回

  • 与 input 具有相同类型的 num_replicas Tensor 对象的列表。

此操作保存来自 tpu.replicate() 计算子图的复制输出。每个复制的输出与输入具有相同的形状和类型。

例如：

%computation = "tf.Computation"()
%replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)

上述计算具有两个副本的复制输出。

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.raw_ops.TPUReplicatedOutput。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。