张量 mask须跟input张量有相同数量的元素数目,但形状或维度不需要相同
参考链接:
https://blog.csdn.net/SoftPoeter/article/details/81667810
torch.masked()理解
最新推荐文章于 2023-11-24 19:21:52 发布
张量 mask须跟input张量有相同数量的元素数目,但形状或维度不需要相同
参考链接:
https://blog.csdn.net/SoftPoeter/article/details/81667810