错误方式
希望在进行softmax之前,如果对被mask掉的位置加上一个特别小的数字,那么softmax之后就会变成0。
pad_mask = (1 - doc_token_mask) * (-1999999) # 把原本0的位置变成一个特别小的数字
qk = qk + pad_mask # 加到原来的上面去
qk_softmax = torch.softmax(qk, dim=-1)
但是这样有两个问题:
- 在fp16的情况下,如果自己随便写的数字特别小,会发生inf
- 在计算梯度的时候,如果是加法,会影响梯度计算。
正确方式
qk = qk.masked_fill_(1-doc_token_mask, -float('inf')) # 把原本0的位置直接变成一个特别小的数字,而且 -float('inf')和精度无关
测试:
a=torch.tensor([1,2,3,4]).float()
mask=torch.tensor([1,1,0,0])
b=a.masked_fill_(1-mask, -float('inf')) # tensor([1., 2., -inf, -inf])
torch.softmax(b, dim=0) # tensor([0.2689, 0.7311, 0.0000, 0.0000])
原文地址:http://www.cnblogs.com/carolsun/p/16807755.html
1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长!
2. 分享目的仅供大家学习和交流,请务用于商业用途!
3. 如果你也有好源码或者教程,可以到用户中心发布,分享有积分奖励和额外收入!
4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解!
5. 如有链接无法下载、失效或广告,请联系管理员处理!
6. 本站资源售价只是赞助,收取费用仅维持本站的日常运营所需!
7. 如遇到加密压缩包,默认解压密码为"gltf",如遇到无法解压的请联系管理员!
8. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载
声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性