要理解pytorch的维度其实很简单,从内往外数括号,每一层括号里有几个元素那一维就是几(然后把数倒过来,越里面的维度越后面),例如2x4x3的张量如下:
torch.tensor([[1,1,1],[1,1,1],[1,1,1],[1,1,1]],[[1,1,1],[1,1,1],[1,1,1],[1,1,1]]])
要理解pytorch的维度其实很简单,从内往外数括号,每一层括号里有几个元素那一维就是几(然后把数倒过来,越里面的维度越后面),例如2x4x3的张量如下:
torch.tensor([[1,1,1],[1,1,1],[1,1,1],[1,1,1]],[[1,1,1],[1,1,1],[1,1,1],[1,1,1]]])