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壓縮文件密碼,文中通過示例代碼介紹的非常詳細(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包的簡單引用方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
Django Admin中增加導(dǎo)出CSV功能過程解析
這篇文章主要介紹了Django Admin中增加導(dǎo)出CSV功能過程解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-09-09
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é)程返回值的四種方法的示例代碼，文中的代碼詳細(xì)易懂，對我們學(xué)習(xí)Python有一定的幫助，需要的朋友可以了解一下2021-12-12
Python3 列表 序列是Python中最基本的數(shù)據(jù)結(jié)構(gòu)，下面這篇文章主要給大家介紹了關(guān)于Python3學(xué)習(xí)筆記之列表方法的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友可以參考下。2017-10-10