pytorch学习记录2--tensor维度变化