pytorch中如何快速判断向量的维度

Posted 2022-07-31 猪猪猪猪侠

除了用size()来查看以外，还可以观察向量中[]的数量，如([1,2])就是一维的，而([[1,2]])就是二维的，依次类推，同样的方式也可用在数据结构的学习当中。