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程序设计入门(3)数组的使用
Jun 16 Python
python使用Berkeley DB数据库实例
Sep 26 Python
python 远程统计文件代码分享
May 14 Python
python实现微信远程控制电脑
Feb 22 Python
Python3 关于pycharm自动导入包快捷设置的方法
Jan 16 Python
为何人工智能(AI)首选Python?读完这篇文章你就知道了(推荐)
Apr 06 Python
PYQT5实现控制台显示功能的方法
Jun 25 Python
Python中sys模块功能与用法实例详解
Feb 26 Python
使用keras2.0 将Merge层改为函数式
May 23 Python
关于Kotlin中SAM转换的那些事
Sep 15 Python
python字符串拼接+和join的区别详解
Dec 03 Python
教你用python控制安卓手机
May 13 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
德生1994机评
2021/03/02 无线电
邮箱正则表达式实现代码(针对php)
2013/06/21 PHP
php in_array() 检查数组中是否存在某个值详解
2016/11/23 PHP
php过滤输入操作之htmlentities与htmlspecialchars用法分析
2017/02/17 PHP
小程序微信退款功能实现方法详解【基于thinkPHP】
2019/05/05 PHP
js 日期转换成中文格式的函数
2009/07/07 Javascript
jQuery根据用户电脑是mac还是pc加载对应样式的方法
2015/06/26 Javascript
JavaScript实现自动消除按钮功能的方法
2015/08/05 Javascript
完美实现bootstrap分页查询
2015/12/09 Javascript
nodejs基础应用
2017/02/03 NodeJs
vue.js从安装到搭建过程详解
2017/03/17 Javascript
Bootstrap 3浏览器兼容性问题及解决方案
2017/04/11 Javascript
详解vue.js 开发环境搭建最简单攻略
2017/06/12 Javascript
php main 与 iframe 相互通讯类(js+php同域/跨域)
2017/09/14 Javascript
JS实现十分钟倒计时代码实例
2018/10/18 Javascript
JS中比较两个Object数组是否相等方法实例
2019/11/11 Javascript
微信小程序中网络请求缓存的解决方法
2019/12/29 Javascript
[01:36:57]【09DOTA2第一视角】小骷髅
2014/04/16 DOTA
[01:30]我们共输赢 完美世界城市挑战赛开启全新赛季
2019/04/19 DOTA
在Windows8上的搭建Python和Django环境
2014/07/03 Python
分享Python字符串关键点
2015/12/13 Python
python函数超时自动退出的实操方法
2020/12/28 Python
HTML5 video 上传预览图片视频如何设置、预览视频某秒的海报帧
2018/08/28 HTML / CSS
德国足球商店:OUTFITTER
2019/05/06 全球购物
SneakerStudio英国:最佳运动鞋商店
2019/05/22 全球购物
即兴演讲稿
2014/01/04 职场文书
2014信息技术专业毕业生自我评价
2014/01/17 职场文书
精神文明单位申报材料
2014/05/02 职场文书
中学生教师节演讲稿
2014/09/03 职场文书
会计人员演讲稿
2014/09/11 职场文书
2014年环保局工作总结
2014/12/11 职场文书
教师年度考核个人总结
2015/02/12 职场文书
七年级之开学家长寄语35句
2019/09/05 职场文书
python 管理系统实现mysql交互的示例代码
2021/12/06 Python
关于MybatisPlus配置双数据库驱动连接数据库问题
2022/01/22 Java/Android
《极主夫道》真人电影正式预告 定档6月3日上映
2022/04/05 日漫