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报错的解决就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

基于pytorch padding=SAME的解决方式

python3的map与reduce实例详解
这篇文章主要介绍了Python3中map()、reduce()、filter()的用法详解,本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2022-08-08
python实现进程间通信简单实例
这篇文章主要介绍了python实现进程间通信简单实例,需要的朋友可以参考下
2014-07-07
Python使用爬虫爬取贵阳房价的方法详解
这篇文章主要为大家详细介绍了Python爬虫爬取贵阳房价的方法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
2022-02-02
Python利用subplots_adjust方法解决图表与画布的间距问题
这篇文章主要介绍了如何在使用python 的 matplotlib库绘图时，使用subplots_adjust()方法来调整图表与画布之间的间距，以及图表与图表之间的间距，感兴趣的可以了解一下
2022-04-04
python3 判断列表是一个空列表的方法
今天小编就为大家分享一篇python3 判断列表是一个空列表的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-05-05
Python实现鸡群算法的示例代码
鸡群算法，缩写为CSO(Chicken Swarm Optimization)，尽管具备所谓仿生学的背景，但实质上是粒子群算法的一个变体。本文将利用Python语言实现这一算法，感兴趣的可以了解一下
2022-11-11
python处理数据,存进hive表的方法
今天小编就为大家分享一篇python处理数据,存进hive表的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-07-07
在Python中利用pickle保存变量的实例
今天小编就为大家分享一篇在Python中利用pickle保存变量的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-12-12
使用Python实现Oracle数据库自动巡检程序
这篇文章主要为大家详细介绍了如何创建一个Oracle数据库自动巡检程序,以确保数据库的顺畅运行,感兴趣的小伙伴可以跟随小编一起学习一下
2024-01-01
python设计tcp数据包协议类的例子
今天小编就为大家分享一篇python设计tcp数据包协议类的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-07-07