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

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


将 N 个输入连接到 N-way 复制的 TPU 计算。

用法

tf.raw_ops.TPUReplicatedInput(
    inputs, is_mirrored_variable=False, index=-1, is_packed=False, name=None
)

参数

  • inputs 至少 1 个具有相同类型的 Tensor 对象的列表。
  • is_mirrored_variable 可选的 bool 。默认为 False 。
  • index 可选的 int 。默认为 -1 。
  • is_packed 可选的 bool 。默认为 False 。
  • name 操作的名称(可选)。

返回

  • 一个Tensor。具有与 inputs 相同的类型。

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

例如：

%a = "tf.opA"()
%b = "tf.opB"()
%replicated_input = "tf.TPUReplicatedInput"(%a, %b)
%computation = "tf.Computation"(%replicated_input)

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

相关用法


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