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的元素個數(shù)需和本tensor相同，但尺寸可以不同

以上這篇pytorch masked_fill報錯的解決就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Django 后臺帶有字典的列表數(shù)據(jù)與頁面js交互實(shí)例
這篇文章主要介紹了Django 后臺帶有字典的列表數(shù)據(jù)與頁面js交互實(shí)例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-04-04
如何使用Python破解ZIP或RAR壓縮文件密碼
這篇文章主要介紹了如何使用Python破解ZIP或RAR壓縮文件密碼,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2020-01-01
8種用Python實(shí)現(xiàn)線性回歸的方法對比詳解
這篇文章主要介紹了8種用Python實(shí)現(xiàn)線性回歸的方法對比詳解，說到如何用Python執(zhí)行線性回歸，大部分人會立刻想到用sklearn的linear_model，但事實(shí)是，Python至少有8種執(zhí)行線性回歸的方法，sklearn并不是最高效的,需要的朋友可以參考下
2019-07-07
python中對正則表達(dá)式re包的簡單引用方式
這篇文章主要介紹了python中對正則表達(dá)式re包的簡單引用方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2022-02-02
Django Admin中增加導(dǎo)出CSV功能過程解析
這篇文章主要介紹了Django Admin中增加導(dǎo)出CSV功能過程解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2019-09-09
Python學(xué)習(xí)之字典和集合的使用詳解
Python 中字典和集合也是非常相近的概念，而且從外觀看上去，也有那么一絲絲的相似。本文將從字典基本知識入手，為你展開字典類型數(shù)據(jù)的各個知識點(diǎn)，需要的可以參考一下
2022-11-11
Python三目運(yùn)算符(三元運(yùn)算符)用法詳解(含實(shí)例代碼)
三元運(yùn)算符在Python里被稱為條件表達(dá)式,這些表達(dá)式基于真(true)/假(false)的條件判斷,在Python 2.4以上才有了三元操作,下面這篇文章主要給大家介紹了關(guān)于Python三目運(yùn)算符(三元運(yùn)算符)用法的相關(guān)資料,需要的朋友可以參考下
2023-02-02
Python腳本在Appium庫上對移動應(yīng)用實(shí)現(xiàn)自動化測試
這篇文章主要介紹了使用Python的Appium庫對移動應(yīng)用實(shí)現(xiàn)自動化測試的教程,屬于Python腳本的一個自動化應(yīng)用,需要的朋友可以參考下
2015-04-04
Python獲取協(xié)程返回值的四種方式詳解
這篇文章主要為大家介紹了Python中獲取協(xié)程返回值的四種方法的示例代碼，文中的代碼詳細(xì)易懂，對我們學(xué)習(xí)Python有一定的幫助，需要的朋友可以了解一下
2021-12-12
Python3學(xué)習(xí)筆記之列表方法示例詳解
Python3 列表序列是Python中最基本的數(shù)據(jù)結(jié)構(gòu)，下面這篇文章主要給大家介紹了關(guān)于Python3學(xué)習(xí)筆記之列表方法的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友可以參考下。
2017-10-10