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之import机制详解
Jul 03 Python
Python 检查数组元素是否存在类似PHP isset()方法
Oct 14 Python
python执行子进程实现进程间通信的方法
Jun 02 Python
在Python的Django框架中生成CSV文件的方法
Jul 22 Python
python实现人民币大写转换
Jun 20 Python
Django配置celery(非djcelery)执行异步任务和定时任务
Jul 16 Python
python3实现爬取淘宝美食代码分享
Sep 23 Python
Numpy之random函数使用学习
Jan 29 Python
Python面向对象程序设计构造函数和析构函数用法分析
Apr 12 Python
Django之form组件自动校验数据实现
Jan 14 Python
python3.8.3安装教程及环境配置的详细教程(64-bit)
Nov 28 Python
如何用python写个模板引擎
Jan 14 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 return语句的另一个作用
2014/07/30 PHP
php中字符串和正则表达式详解
2014/10/23 PHP
在PHP中输出JS语句以及乱码问题的解决方案
2019/02/13 PHP
js实现addClass,removeClass,hasClass的函数代码
2011/07/13 Javascript
Javascript 闭包引起的IE内存泄露分析
2012/05/23 Javascript
JQUERY 实现窗口滚动搜索框停靠效果(类似滚动停靠)
2013/03/27 Javascript
JQuery下拉框应用示例介绍
2014/04/23 Javascript
jQuery设置和移除文本框默认值的方法
2015/03/09 Javascript
一道关于JavaScript变量作用域的面试题
2016/03/08 Javascript
基于React.js实现原生js拖拽效果引发的思考
2016/03/30 Javascript
jQuery基础的工厂函数以及定时器的经典实例分析
2016/05/20 Javascript
jQuery解决input元素的blur事件和其他非表单元素的click事件冲突问题
2016/08/15 Javascript
AngularJS中watch监听用法分析
2016/11/04 Javascript
JS异步加载的三种实现方式
2017/03/16 Javascript
解决vue2.x中数据渲染以及vuex缓存的问题
2017/07/13 Javascript
jQuery remove()过滤被删除的元素(推荐)
2017/07/18 jQuery
微信小程序使用Socket的实例
2017/09/19 Javascript
解决betterScroll在vue中存在图片时,出现拉不动的问题
2018/09/27 Javascript
vue自定义树状结构图的实现方法
2020/10/18 Javascript
vue2.0 watch里面的 deep和immediate用法说明
2020/10/30 Javascript
[00:06]Yes,it worked!小卡尔成功穿越时空加入战场!
2019/07/20 DOTA
[48:18]DOTA2-DPC中国联赛 正赛 RNG vs Dynasty BO3 第二场 1月29日
2021/03/11 DOTA
Python可视化mhd格式和raw格式的医学图像并保存的方法
2019/01/24 Python
微软俄罗斯官方网站:Microsoft俄罗斯
2016/09/18 全球购物
万得城电器土耳其网站:欧洲第一大电子产品零售商
2016/10/07 全球购物
The North Face北面德国官网:美国著名户外品牌
2018/12/12 全球购物
办公文员的工作岗位职责
2013/11/12 职场文书
党建工作经验交流材料
2014/05/25 职场文书
市级三好学生事迹材料
2014/08/27 职场文书
县长群众路线对照检查材料思想汇报
2014/10/02 职场文书
中国合伙人观后感
2015/06/02 职场文书
董事长致辞
2015/07/29 职场文书
基于Redis位图实现用户签到功能
2021/05/08 Redis
教你使用Python pypinyin库实现汉字转拼音
2021/05/27 Python
Nginx反向代理学习实例教程
2021/10/24 Servers
WebRTC记录音视频流(web技术分享)
2022/02/24 Javascript