pytorch masked_fill报错的解决
更新时间：2020年02月18日 08:48:48 作者：昕晴
今天小编就为大家分享一篇pytorch masked_fill报错的解决，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
如下所示：
import torch.nn.functional as F import numpy as np a = torch.Tensor([1,2,3,4]) a = a.masked_fill(mask = torch.ByteTensor([1,1,0,0]), value=-np.inf) print(a) b = F.softmax(a) print(b)
tensor([-inf, -inf, 3., 4.])
d:/pycharmdaima/star-transformer/ceshi.py:8: UserWarning: Implicit dimension choice for softmax has been deprecated. Change
the call to include dim=X as an argument.
b = F.softmax(a)
tensor([0.0000, 0.0000, 0.2689, 0.7311])
容易报错：
Expected object of scalar type Byte but got scalar type Long for argument #2 'mask'
原因，
mask = torch.LongTensor()
解决方法：
mask = torch.ByteTensor()
在mask值为1的位置处用value填充。mask的元素个数需和本tensor相同，但尺寸可以不同
以上这篇pytorch masked_fill报错的解决就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家详细介绍了Python进程,线程和协程，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-03-03
本文介绍如何使用 Python 的 smtplib 和 email 库通过 QQ 邮箱发送电子邮件,文中有详细的代码示例供大家参考,具有一定的参考价值,需要的朋友可以参考下2023-10-10
今天小编就为大家分享一篇Python项目 基于Scapy实现SYN泛洪攻击的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
这篇文章主要为大家介绍了Cpython的GIL解释器锁工作机制掌握示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-07-07
这篇文章主要为大家详细介绍了python之DataFrame实现excel合并单元格，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-04-04
本文为大家总结了五款利用Python+Pygame实现的童年经典游戏：推箱子、滑雪、八分音符酱、保卫萝卜和飞机大战，快跟随小编一起学习一下2022-04-04
这篇文章主要介绍了python 时间处理之月份加减问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-11-11
这篇文章主要介绍了Python文件操作基础流程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03
这篇文章主要介绍了matplotlib 对坐标的控制,加图例注释的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-04-04
这篇文章主要为大家介绍了如何在Matplotlib中使用LaTeX 公式和符号以及Python如何生成LaTeX数学公式。文中的示例代码讲解详细，感兴趣的小伙伴可以了解一下2022-02-02
最新评论