ttnn.dump_tensor

ttnn.dump_tensor(file_name: str | pathlib.Path, tensor: ttnn.Tensor) → `None`

Dump tensor to a file.

Parameters:

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

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

Returns:

None – tensor saved to a specified file.

Example

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