u013630299的博客

1、torch.matmul（）就是简单的矩阵乘法2、torch.permute(input,dims) ，按照指定的维度对tensor的重新排列>>> x = torch.randn(2, 3, 5)>>> x.size()torch.Size([2, 3, 5])>>> torch.permute(x, (2, 0, 1)).size()torch.Size([5, 2, 3]3、torch.unsqueeze(input,..