當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.distribute.experimental.partitioners.FixedShardsPartitioner用法及代碼示例


分配固定數量分片的分區器。

繼承自：Partitioner

用法

tf.distribute.experimental.partitioners.FixedShardsPartitioner(
    num_shards
)

參數

  • num_shards int ，要分區的分片數。

例子：

# standalone usage:
partitioner = FixedShardsPartitioner(num_shards=2)
partitions = partitioner(tf.TensorShape([10, 3]), tf.float32)
[2, 1]

# use in ParameterServerStrategy
# strategy = tf.distribute.experimental.ParameterServerStrategy(
#   cluster_resolver=cluster_resolver, variable_partitioner=partitioner)

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.distribute.experimental.partitioners.FixedShardsPartitioner。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。