Giters

NVIDIA / MatX

An efficient C++17 GPU numerical computing library with Python-like syntax

Home Page:https://nvidia.github.io/MatX

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[FEA] Allow io::WriteMAT to write multiple tensors.

nvjonwong opened this issue · comments

commented

Currently io::WriteMAT can only write one tensor per file.

Proposal: Allow users to add to a pybind11::dict and then add tensors to that. Provide WriteMAT overload that takes a pybind11::dict directly and writes it to file.

https://github.com/NVIDIA/MatX/tree/nvjonwong/write_multiple_to_mat