pytorch 获取tensor维度信息