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

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。