DTensor

class tensorplay.distributed.tensor.DTensor(local_tensor: Any, device_mesh: DeviceMesh, placements: Sequence[Placement], *, shape: Sequence[int] | None = None, stride: Sequence[int] | None = None, grad_placements: Sequence[Placement] | None = None, backward_dtype: Any = None)

A logical tensor represented by a local value and a mesh placement.