pytorch masked_fill报错的解决
更新时间：2020年02月18日 08:48:48 作者：昕晴
今天小编就为大家分享一篇pytorch masked_fill报错的解决，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
如下所示：
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报错的解决就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Django 后台带有字典的列表数据与页面js交互实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-04-04
这篇文章主要介绍了如何使用Python破解ZIP或RAR压缩文件密码,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
这篇文章主要介绍了8种用Python实现线性回归的方法对比详解，说到如何用Python执行线性回归，大部分人会立刻想到用sklearn的linear_model，但事实是，Python至少有8种执行线性回归的方法，sklearn并不是最高效的,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了python中对正则表达式re包的简单引用方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-02-02
这篇文章主要介绍了Django Admin中增加导出CSV功能过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-09-09
Python 中字典和集合也是非常相近的概念，而且从外观看上去，也有那么一丝丝的相似。本文将从字典基本知识入手，为你展开字典类型数据的各个知识点，需要的可以参考一下2022-11-11
三元运算符在Python里被称为条件表达式,这些表达式基于真(true)/假(false)的条件判断,在Python 2.4以上才有了三元操作,下面这篇文章主要给大家介绍了关于Python三目运算符(三元运算符)用法的相关资料,需要的朋友可以参考下2023-02-02
这篇文章主要介绍了使用Python的Appium库对移动应用实现自动化测试的教程,属于Python脚本的一个自动化应用,需要的朋友可以参考下2015-04-04
这篇文章主要为大家介绍了Python中获取协程返回值的四种方法的示例代码，文中的代码详细易懂，对我们学习Python有一定的帮助，需要的朋友可以了解一下2021-12-12
Python3 列表 序列是Python中最基本的数据结构，下面这篇文章主要给大家介绍了关于Python3学习笔记之列表方法的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友可以参考下。2017-10-10
最新评论