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脚本实现分析dns日志并对受访域名排行
Sep 18 Python
python实现多线程暴力破解登陆路由器功能代码分享
Jan 04 Python
全面理解Python中self的用法
Jun 04 Python
Python实现将数据库一键导出为Excel表格的实例
Dec 30 Python
Windows环境下python环境安装使用图文教程
Mar 13 Python
Python常用字符串替换函数strip、replace及sub用法示例
May 21 Python
Django ORM 聚合查询和分组查询实现详解
Aug 09 Python
10个Python面试常问的问题(小结)
Nov 20 Python
python+opencv实现移动侦测(帧差法)
Mar 20 Python
使用python-Jenkins批量创建及修改jobs操作
May 12 Python
解决Python Matplotlib绘图数据点位置错乱问题
May 16 Python
python怎么提高计算速度
Jun 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
Yii中srbac权限扩展模块工作原理与用法分析
2016/07/14 PHP
thinkPHP批量删除的实现方法分析
2016/11/09 PHP
php 判断页面或图片是否经过gzip压缩的方法
2017/04/05 PHP
番茄的表单验证类代码修改版
2008/07/18 Javascript
利用jQuery实现可以编辑的表格
2014/05/26 Javascript
js实例属性和原型属性示例详解
2014/11/23 Javascript
分享2个jQuery插件--jquery.fileupload与artdialog
2014/12/26 Javascript
jQuery中parentsUntil()方法用法实例
2015/01/07 Javascript
jQuery实现行文字链接提示效果的方法
2015/03/10 Javascript
基于jQuery实现的向下滑动二级菜单效果代码
2015/08/31 Javascript
Node.js环境下编写爬虫爬取维基百科内容的实例分享
2016/06/12 Javascript
vue2.0使用Sortable.js实现的拖拽功能示例
2017/02/21 Javascript
webpack打包单页面如何引用的js
2017/06/07 Javascript
JavaScript表格隔行变色和Tab标签页特效示例【附jQuery版】
2019/07/11 jQuery
js实现点赞按钮功能的实例代码
2020/03/06 Javascript
OpenLayers加载缩放控件使用方法详解
2020/09/25 Javascript
python实现从字符串中找出字符1的位置以及个数的方法
2014/08/25 Python
Python中计算三角函数之cos()方法的使用简介
2015/05/15 Python
wxpython绘制音频效果
2019/11/18 Python
python中使用input()函数获取用户输入值方式
2020/05/03 Python
米兰必去买手店排行榜首位:Antonioli
2016/09/11 全球购物
英国香水店:The Perfume Shop
2017/03/27 全球购物
英国最大的专业户外零售商:Mountain Warehouse
2018/06/06 全球购物
澳大利亚汽车零部件、音响及配件超市:Automotive Superstore
2018/06/19 全球购物
巴西补充剂和维生素购物网站:Natue
2019/06/17 全球购物
Keds加拿大官网:购买帆布运动鞋和皮鞋
2019/09/26 全球购物
美团网旗下网上订餐平台:美团外卖
2020/03/05 全球购物
Java中的类包括什么内容?设计时要注意哪些方面
2012/05/23 面试题
工作会议欢迎词
2014/01/16 职场文书
网吧消防安全制度
2014/01/28 职场文书
军训教官感言
2014/03/02 职场文书
大学新生军训自我鉴定
2014/03/18 职场文书
教师师德演讲稿
2014/05/06 职场文书
网络研修心得体会
2016/01/08 职场文书
创业计划书之书店
2019/09/10 职场文书
导游词之桂林山水
2019/09/20 职场文书