Closed nvjonwong closed 1 year ago
Ideally TensorViewToNumpy should work with any rank and the DLPack support we added. The other way is it as easy since it's going from dynamic ranks to static, and you have to bound it to a point.
This is my attempt at being able to intake more arbitrary numpy data. https://github.com/NVIDIA/MatX/pull/428
Current TensorViewToNumpy and NumpyToTensorViews only work on rank < 5 matrices.
Proposing to increase to add rank=5 support and add a
static_assert
so users will know about the implementation limitation.