ttnn.dump_tensor

ttnn.dump_tensor = Operation(python_fully_qualified_name='ttnn.dump_tensor', function=<function dump_tensor>, preprocess_golden_function_inputs=<function default_preprocess_golden_function_inputs>, golden_function=None, postprocess_golden_function_outputs=<function default_postprocess_golden_function_outputs>, is_cpp_operation=False, is_experimental=False)

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)