1,找峰
热图shape:aaa.shape=[100,100,1]
卷积maxpool后为bbb
ccc=bbb==aaa
和原来点相等,则是峰
2,各种矩阵计算利用einsum
aaa=torch.rand(1,2,3,2)
bbb=torch.rand(2,3,2)
ccc=torch.einsum('ijkg,jlg->iklg',[aaa,bbb])
解释:
‘ijkg,jlg->iklg’ 式(1)
a有ijkg,b有jlg
下文中的词“前面”、“后面”表示->号的前后
(1)一样的后面没有就相加消除。式(1)中前面都有j,后面没j,j就相加消除了。
(2)一样的前面后面都有,不会纳入计算。可以认为是element wise 计算
(3)不一样的前面后面都有,重复计算。类似矩阵计算那样每个元素都加乘一遍