pytorch 常用函數(shù)參數(shù)詳解

1、torch.cat(inputs, dim=0) -> Tensor?

參考鏈接:

[Pytorch] 詳解 torch.cat()

Pytorch學習筆記(一):torch.cat()模塊的詳解

函數(shù)作用:cat 是 concatnate 的意思:拼接徘熔,聯(lián)系在一起募寨。在給定維度上對輸入的 Tensor 序列進行拼接操作寄月。torch.cat 可以看作是 torch.split 和 torch.chunk 的反操作

參數(shù):

inputs(sequence of Tensors):可以是任意相同類型的 Tensor 的 python 序列

dim(int, optional):defaults=0

dim=0: 按列進行拼接?

dim=1: 按行進行拼接

dim=-1: 如果行和列數(shù)都相同則按行進行拼接酵紫,否則按照行數(shù)或列數(shù)相等的維度進行拼接

假設 a 和 b 都是 Tensor,且 a 的維度為 [2, 3]河劝,b 的維度為 [2, 4]惠啄,則

torch.cat((a, b), dim=1) 的維度為 [2, 7]


2慎恒、torch.nn.CrossEntropyLoss()

函數(shù)作用:CrossEntropy 是交叉熵的意思,故而 CrossEntropyLoss 的作用是計算交叉熵撵渡。CrossEntropyLoss 函數(shù)是將 torch.nn.Softmax 和 torch.nn.NLLLoss 兩個函數(shù)組合在一起使用融柬,故而傳入的預測值不需要先進行 torch.nnSoftmax 操作。

參數(shù):

input(N, C):N 是 batch_size趋距,C 則是類別數(shù)粒氧,即在定義模型輸出時,輸出節(jié)點個數(shù)要定義為 [N, C]节腐。其中特別注意的是 target 的數(shù)據(jù)類型需要是浮點數(shù)外盯,即 float32

target(N):N 是 batch_size,故 target 需要是 1D 張量翼雀。其中特別注意的是 target 的數(shù)據(jù)類型需要是 long饱苟,即 int64

例子:

loss = nn.CrossEntropyLoss()

input = torch.randn(3, 5, requires_grad=True, dtype=torch.float32)

target = torch.empty(3, dtype=torch.long).random_(5)

output = loss(input, target)

output

輸出為:

tensor(1.6916, grad_fn=<NllLossBackward>)

?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市狼渊,隨后出現(xiàn)的幾起案子箱熬,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,386評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件城须,死亡現(xiàn)場離奇詭異蚤认,居然都是意外死亡,警方通過查閱死者的電腦和手機糕伐,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,142評論 3 394
  • 文/潘曉璐 我一進店門砰琢,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人赤炒,你說我怎么就攤上這事氯析。” “怎么了莺褒?”我有些...
    開封第一講書人閱讀 164,704評論 0 353
  • 文/不壞的土叔 我叫張陵,是天一觀的道長雪情。 經(jīng)常有香客問我遵岩,道長,這世上最難降的妖魔是什么巡通? 我笑而不...
    開封第一講書人閱讀 58,702評論 1 294
  • 正文 為了忘掉前任尘执,我火速辦了婚禮,結果婚禮上宴凉,老公的妹妹穿的比我還像新娘誊锭。我一直安慰自己,他們只是感情好弥锄,可當我...
    茶點故事閱讀 67,716評論 6 392
  • 文/花漫 我一把揭開白布丧靡。 她就那樣靜靜地躺著,像睡著了一般籽暇。 火紅的嫁衣襯著肌膚如雪温治。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,573評論 1 305
  • 那天戒悠,我揣著相機與錄音熬荆,去河邊找鬼。 笑死绸狐,一個胖子當著我的面吹牛卤恳,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播寒矿,決...
    沈念sama閱讀 40,314評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼突琳,長吁一口氣:“原來是場噩夢啊……” “哼噩死!你這毒婦竟也來了沮明?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,230評論 0 276
  • 序言:老撾萬榮一對情侶失蹤仙蛉,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后冠息,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體挪凑,經(jīng)...
    沈念sama閱讀 45,680評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,873評論 3 336
  • 正文 我和宋清朗相戀三年逛艰,在試婚紗的時候發(fā)現(xiàn)自己被綠了躏碳。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 39,991評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡散怖,死狀恐怖菇绵,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情镇眷,我是刑警寧澤咬最,帶...
    沈念sama閱讀 35,706評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站欠动,受9級特大地震影響永乌,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜具伍,卻給世界環(huán)境...
    茶點故事閱讀 41,329評論 3 330
  • 文/蒙蒙 一翅雏、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧人芽,春花似錦望几、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,910評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至祈坠,卻和暖如春害碾,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背赦拘。 一陣腳步聲響...
    開封第一講書人閱讀 33,038評論 1 270
  • 我被黑心中介騙來泰國打工慌随, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人躺同。 一個月前我還...
    沈念sama閱讀 48,158評論 3 370
  • 正文 我出身青樓阁猜,卻偏偏與公主長得像,于是被迫代替她去往敵國和親蹋艺。 傳聞我的和親對象是個殘疾皇子剃袍,可洞房花燭夜當晚...
    茶點故事閱讀 44,941評論 2 355

推薦閱讀更多精彩內(nèi)容