One-line tensor visualization for PyTorch and NumPy

Maybe an example notebook that saves a demo image that could be included in the README?