pytorch中与维度/变换相关的几个函数

torch.size () 先说torch.size()函数，由于后面的方法都会用这个方法看到变换后的矩阵的维度 经过该方法，能够查看当前Tensor的维度，用法也很简单： >>>import torch >>>a = torch.Tensor([[[1, 2, 3], [4, 5, 6]]]) >>>a.size() torch.Size([1, 2, 3]) torch.view() 官方文档

>>阅读原文<<