ttnn.dump_tensor

ttnn.dump_tensor(file_name: str | pathlib.Path, tensor: ttnn.Tensor, distribute: Dict[str, str] | None = None) → `None`

Dump tensor to a file.

Parameters:

  • file_name (str | pathlib.Path) – The file name.

  • tensor (ttnn.Tensor) – the tensor to be dumped.

  • distribute (Dict[str, str], optional) – The distributed strategy. Only applicable to multi-device tensors. Defaults to None.

Returns:

None – tensor saved to a specified file.

Example

>>> tensor = ttnn.ones([2, 3], bfloat16, ttnn.ROW_MAJOR_LAYOUT)
>>> dump_tensor(file_name=str(tensor.bin), tensor=tensor)