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

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


用法

__call__(
    shape, dtype, axis=0
)

参数

  • shape a tf.TensorShape ，要分区的形状。
  • dtype tf.dtypes.Dtype 指示分区值的类型。
  • axis 要分割的轴。默认值：最外轴。

返回

  • 表示每个轴上的分区数的整数列表，其中 i-th 值对应于 i-th 轴。

对给定的shape 进行分区并返回分区结果。

分配固定数量分片的分区器示例：

partitioner = FixedShardsPartitioner(num_shards=2)
partitions = partitioner(tf.TensorShape([10, 3], tf.float32), axis=0)
print(partitions) # [2, 0]

相关用法


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