ttnn.load_tensor

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

Load tensor from a file.

Parameters:

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

Keyword Arguments:

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

Returns:

ttnn.Tensor – the loaded tensor.

Example

>>> device = ttnn.open_device(0)
>>> tensor = ttnn.load_tensor(file_name=str(tensor.bin), device=device)