ttnn.get_device_tensors

ttnn.get_device_tensors = <nanobind.nb_func object>

ttnn._ttnn.tensor.Tensor) -> list[ttnn._ttnn.tensor.Tensor]

Get a list of tensor shards from a multidevice tensor.

Parameters:

tensor (Tensor) – The tensor to get the shards from.

Returns:

List[Tensor] – The shards of the tensor corresponding to the devices.

Type:

get_device_tensors(tensor