Shard

class paddle.distributed. Shard

描述 Tensor 在多设备间按照指定的维度来切分张量。

参数

  • dim (int) - 指定张量的切分维度。

代码示例

COPY-FROM: paddle.distributed.Shard