牛逼代码记录

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）不一样的前面后面都有，重复计算。类似矩阵计算那样每个元素都加乘一遍