pytorch运算操作

torch.transponse(dim1,dim2)

a = torch.rand((2,3,10))
b = torch.rand((2,2,10))

res1 = torch.matmul(a,b.transpose(1,2))
#res1 = torch.matmul(a,b.transose(1,2))
print res1.size()

output:
    [torch.FloatTensor of size 2x3x2]