ttnn.load_tensor

ttnn.load_tensor(file_name: str | pathlib.Path, *, device: ttnn.MeshDevice = None) → ttnn.Tensor

Load tensor from a file.

Parameters:

file_name (str | pathlib.Path) – the file name.

Keyword Arguments:

device (ttnn.MeshDevice, optional) – the device. Defaults to None.

Returns:

ttnn.Tensor – the loaded tensor.

Example

# Create a TT-NN tensor, dump its contents, and load it back
ttnn_tensor = ttnn.rand((2, 3), dtype=ttnn.bfloat16, device=device)
ttnn.dump_tensor(file_name="ttnn_tensor.tensorbin", tensor=ttnn_tensor)

loaded_tensor = ttnn.load_tensor(file_name="ttnn_tensor.tensorbin", device=device)
logger.info("Loaded TT-NN tensor shape", loaded_tensor.shape)