tensorplay.distributed.tensor.distribute_tensor

tensorplay.distributed.tensor.distribute_tensor(tensor: Any, device_mesh: DeviceMesh | None = None, placements: Sequence[Placement] | None = None, *, src_data_rank: int | None = 0) → DTensor[source]