Tensor visualizer

I often wish there were a decent visualizer for NumPy/PyTorch tensors. Debugging dimension indexing bugs with the way the Python extension displays tensors is a total drag :(

I know that Debug Visualizer has grid visualization, but that only works for 2D tensors with several hundred elements. I routinely need to deal with 4+ dimensional tensors with millions of items, so a simple grid doesn't cut it. There would need to be some sort of summarization along large dimensions and probably a UI for choosing which dimensions to display.

I know, this is quite a bit to ask, but if Debug Visualizer had that, I suspect it would be a lot more popular with data scientists!