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递归计算N!的方法
May 05 Python
浅谈编码,解码,乱码的问题
Dec 30 Python
Python 获取当前所在目录的方法详解
Aug 02 Python
Python正则表达式非贪婪、多行匹配功能示例
Aug 08 Python
Python字典中的键映射多个值的方法(列表或者集合)
Oct 17 Python
python实现批量视频分帧、保存视频帧
May 31 Python
Python编译成.so文件进行加密后调用的实现
Dec 23 Python
Python常用库Numpy进行矩阵运算详解
Jul 21 Python
如何基于python把文字图片写入word文档
Jul 31 Python
python归并排序算法过程实例讲解
Nov 04 Python
Python爬虫爬取微博热搜保存为 Markdown 文件的源码
Feb 22 Python
Python实现Excel自动分组合并单元格
Feb 22 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
对text数据类型不支持代码页转换 从: 1252 到: 936
2011/04/23 PHP
刷新时清空文本框内容的js代码
2007/04/23 Javascript
artDialog 4.1.5 Dreamweaver代码提示/补全插件 附下载
2012/07/31 Javascript
js获取当前路径的简单示例代码
2014/01/08 Javascript
JavaScript中遍历对象的property的3种方法介绍
2014/12/30 Javascript
jQuery实现动态添加和删除一个div
2015/08/12 Javascript
js查看一个函数的执行时间实例代码
2015/09/12 Javascript
JS基于Mootools实现的个性菜单效果代码
2015/10/21 Javascript
在其他地方你学不到的jQuery小贴士和技巧(欢迎收藏)
2016/01/20 Javascript
javascript每日必学之封装
2016/02/23 Javascript
javascript 继承学习心得总结
2016/03/17 Javascript
JavaScript数据绑定实现一个简单的 MVVM 库
2016/04/08 Javascript
JS 动态加载js文件和css文件 同步/异步的两种简单方式
2016/09/23 Javascript
JS在浏览器中解析Base64编码图像
2017/02/09 Javascript
基于bootstrap实现收缩导航条
2017/03/17 Javascript
详解Angular2 之 结构型指令
2017/06/21 Javascript
Vue之Watcher源码解析(2)
2017/07/19 Javascript
vue打包静态资源后显示空白及static文件路径报错的解决
2020/09/02 Javascript
python爬虫实战之最简单的网页爬虫教程
2017/08/13 Python
python远程连接MySQL数据库
2019/04/19 Python
Python微信操控itchat的方法
2019/05/31 Python
Python数据库小程序源代码
2019/09/15 Python
Django的ListView超详细用法(含分页paginate)
2020/05/21 Python
python 基于DDT实现数据驱动测试
2021/02/18 Python
CSS3的resize属性使用初探
2015/09/27 HTML / CSS
用html5的canvas画布绘制贝塞尔曲线完整代码
2013/08/14 HTML / CSS
html5手机端页面可以向右滑动导致样式受影响的问题
2018/06/20 HTML / CSS
Exoticca英国:以最优惠的价格提供豪华异国情调旅行
2018/10/18 全球购物
物业管理公司实习生自我鉴定
2013/09/19 职场文书
大学生自我鉴定范文模板
2014/01/21 职场文书
2014年测量员工作总结
2014/12/12 职场文书
长城英文导游词
2015/01/30 职场文书
财务个人年度总结范文
2015/02/26 职场文书
试用期辞职信范文
2015/03/02 职场文书
2015年英语教师工作总结
2015/05/20 职场文书
python中的被动信息搜集
2021/04/29 Python