pytorch 四维数组可视化