0%

pytorch运算操作

发表于 更新于 分类于 阅读次数： Valine：
本文字数： 97 阅读时长 ≈ 1 分钟

1.transponse

​ torch.transponse操作是转置操作，是一种在矩阵乘法中最常用的几种操作之一。

1
2
#交换两个维度
torch.transponse(dim1,dim2)

2.matmul和bmm

​ matmul和bmm都是实现batch的矩阵乘法操作。

1
2
3
4
5
6
7
8
9
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]