torch_frame.utils.save

save(tensor_frame: TensorFrame, col_stats: dict[str, dict[StatType, Any]] | None, path: str)

保存一个TensorFrame对象和可选的col_stats到指定路径。

Parameters:

• tensor_frame (TensorFrame) – 要保存的 TensorFrame 对象。

• col_stats (dict, 可选) – 要保存的col_stats。

• path (str) – 保存 TensorFrame 对象和 col_stats 的路径。