pytorch masked_fill报错的解决


Posted in Python onFebruary 18, 2020

如下所示:

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 相关文章推荐
python统计字符串中指定字符出现次数的方法
Apr 04 Python
python结合selenium获取XX省交通违章数据的实现思路及代码
Jun 26 Python
Python 制作糗事百科爬虫实例
Sep 22 Python
Pycharm学习教程(7)虚拟机VM的配置教程
May 04 Python
Pyinstaller将py打包成exe的实例
Mar 31 Python
Python django框架应用中实现获取访问者ip地址示例
May 17 Python
Python实现二叉树前序、中序、后序及层次遍历示例代码
May 18 Python
使用python画社交网络图实例代码
Jul 10 Python
python+selenium 脚本实现每天自动登记的思路详解
Mar 11 Python
python实现猜数游戏(保存游戏记录)
Jun 22 Python
详细分析Python可变对象和不可变对象
Jul 09 Python
python 下载m3u8视频的示例代码
Nov 11 Python
python分别打包出32位和64位应用程序
Feb 18 #Python
pandas和spark dataframe互相转换实例详解
Feb 18 #Python
PyTorch中Tensor的数据统计示例
Feb 17 #Python
pytorch实现Tensor变量之间的转换
Feb 17 #Python
Macbook安装Python最新版本、GUI开发环境、图像处理、视频处理环境详解
Feb 17 #Python
PyCharm无法识别PyQt5的2种解决方法,ModuleNotFoundError: No module named 'pyqt5'
Feb 17 #Python
python识别验证码图片实例详解
Feb 17 #Python
You might like
深入PHP运行环境配置的详解
2013/06/04 PHP
php遍历删除整个目录及文件的方法
2015/03/13 PHP
php压缩和解压缩字符串的方法
2015/03/14 PHP
php随机生成数字字母组合的方法
2015/03/18 PHP
CodeIgniter删除和设置Cookie的方法
2015/04/07 PHP
分享别人写的一个小型js框架
2007/08/13 Javascript
javascript 仿QQ滑动菜单效果代码
2010/09/03 Javascript
JavaScript内核之基本概念
2011/10/21 Javascript
javascript学习笔记(十七) 检测浏览器插件代码
2012/06/20 Javascript
JavaScript动态创建div属性和样式示例代码
2013/10/09 Javascript
推荐4个原生javascript常用的函数
2015/01/12 Javascript
jquery实现从数组移除指定的值
2015/06/24 Javascript
jQuery实现的个性化返回底部与返回顶部特效代码
2015/10/30 Javascript
AngularJS利用Controller完成URL跳转
2016/08/09 Javascript
js面向对象编程总结
2017/02/16 Javascript
js 函数式编程学习笔记
2017/03/25 Javascript
使用pm2部署node生产环境的方法步骤
2019/03/09 Javascript
js实现动态时钟
2020/03/12 Javascript
Vue实现简单的留言板
2020/10/23 Javascript
js用正则表达式筛选年月日的实例方法
2021/01/04 Javascript
Python聊天室实例程序分享
2016/01/05 Python
python实现多线程的方式及多条命令并发执行
2016/06/07 Python
解决pandas无法在pycharm中使用plot()方法显示图像的问题
2018/05/24 Python
Python设置在shell脚本中自动补全功能的方法
2018/06/25 Python
基于python及pytorch中乘法的使用详解
2019/12/27 Python
python主要用于哪些方向
2020/07/05 Python
使用html5 canvas 画时钟代码实例分享
2015/11/11 HTML / CSS
英国音乐设备和乐器商店:Gear4music
2017/10/16 全球购物
德国孕妇装和婴童服装网上商店:bellybutton
2018/04/12 全球购物
俄罗斯奢侈品牌衣服、鞋子和配饰的在线商店:INTERMODA
2020/07/17 全球购物
特岗教师个人总结
2015/02/10 职场文书
安全员岗位职责范本
2015/04/11 职场文书
紧急通知
2015/04/17 职场文书
劳动仲裁撤诉申请书
2015/05/18 职场文书
Python进行区间取值案例讲解
2021/08/02 Python
Tomcat执行startup.bat出现闪退的原因及解决办法
2022/04/20 Servers