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

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__。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。