當前位置:偏方大全网 - 偏方秘方 - Pytorch火炬級

Pytorch火炬級

本文介紹了torch類中壹些可能有用的方法。

輸入被拼接,參數指示要拼接的尺寸。

沿給定軸dim在輸入索引張量index指定的位置賦值。

賦值公式如下:

也就是說,修改輸入尺寸的數值排列。

示例:

根據mask張量中的二進制值,取輸入張量中的指定項(mask是ByteTensor),將值返回給壹個新的1D張量。

即在原始輸入中選擇壹些值。

返回包含輸入input中非零元素索引的張量。輸出張量中的每壹行都包含輸入中非零元素的索引。

從輸入張量形狀中移除1並返回。如果輸入形狀為(a×1×b×1×c×1×D),則輸出形狀為(A×B×C×D)。

示例:

沿新維度對輸入張量進行排序。最外面的維度是0。

示例:

轉置壹個二維向量。

返回壹個新的張量,並為輸入的指定位置插入維度1。如果dim為負,則轉換為dim+input.dim()+1。

設置用於生成隨機數的種子。

返回生成的隨機數的原始種子值。

返回壹個張量,它包含從給定參數均值std的離散正態分布中提取隨機數。

計算輸入張量每個元素的絕對值。

返回包含輸入張量每個元素的反余弦值的新張量。

將標量值逐個元素地加到輸入張量input上,並將結果返回給壹個新的張量out,即out =張量+值。

另壹個張量的每個元素乘以壹個標量值並加到iput張量上。將結果返回到輸出張量out。即out=input+(其他?值).

向上舍入輸入張量每個元素。

向下舍入。

將輸入張量的每個元素鉗制到區間[min,max],並將結果返回到新的張量。

將輸入逐元素除以標量值,並將結果返回到輸出張量out。

對有起始和終止的兩個張量做線性插值,結果返回輸出張量。

計算輸入的自然對數。

將輸入input的每個元素乘以標量值,並返回新的結果張量。

返回包含輸入張量每個元素的倒數的新張量,即1.0/x。

返回壹個新張量,包含輸入張量每個元素的除法余數。

四舍五入。

計算每個元素平方根的倒數。

輸入輸入張量的每個元素的sigmoid值。

返回指定維度上輸入的累積積。比如輸入是n元向量,結果也是n元向量,第I個輸出元素的值是yi=x1?x2?x3?...?xi .

返回指定維度上輸入的累積和。第I個輸出元素的值為yi=x1+x2+x3+...+xi。

返回輸入張量所有元素的平均值。

返回輸入張量給定維度中每行的乘積。

示例:

返回輸入張量輸入的所有元素的標準差。

返回輸入張量給定維度中每壹行的總和。輸出形狀與輸入形狀相同,只是在給定的維度中是1。

返回輸入張量所有元素的方差。

將輸入張量與第二個參數進行比較。如果相同,則在相應位置返回1。第二個參數可以是張量或數字。

示例:

如果兩個張量具有相同的形狀和元素值,則返回True,否則返回False。

即比較張量形狀和元素值。

逐元素比較input和其他元素的大小,即input是否>;=其他.

逐元素比較input和其他元素的大小,即input是否>;其他.

逐元素比較輸入和其他元素,即是否輸入

逐元素比較輸入和其他元素,即是否輸入

返回輸入張量所有元素的最大值。

返回輸入張量給定維中每壹行的最大值,同時返回每個最大值的位置索引。

返回輸入張量所有元素的最小值。

將輸入中的元素與另壹個位置的元素逐壹比較,將最小值返回給輸出張量。即outi=min(tensori,otheri)。

沿指定維度以升序對輸入張量輸入進行排序。如果沒有給出尺寸,則默認為輸入的最後壹個尺寸。如果指定的參數descending為真,將按降序排序。

返回壹個元組(sorted _ tensor,sorted _ indicators),其中sorted _ indicators是原始輸入中的下標。

返回沿給定維度的輸入張量輸入中的k個最大值。如果未指定dim,則默認為輸入的最後壹個尺寸。如果minimum為False,則返回最小k值。

返回壹個元組(值,索引),其中索引是原始輸入張量輸入中測量元素的下標。如果布爾值sorted被設置為True,它將確保返回的k值被排序。

返回沿維度dim的兩個張量input和other的叉積。輸入和其他必須具有相同的形狀,並且指定的dim維度的大小必須為3。

對存儲在兩個批次batch1和batch2中的矩陣執行批次矩陣乘法。

矩陣乘法在矩陣mat1和mat2上執行。

out=(beta?M)+(阿爾法?mat1@mat2)

對存儲在兩個批次batch1和batch2中的矩陣執行批量矩陣乘法。batch1和batch2都是包含相同數量矩陣的三維張量。如果batch1是形狀為b×n×m的張量,batch1是形狀為b×m×p的張量,那麽out和mat的形狀都是n×p,即RES =(β?M)+(阿爾法?sum(batch1i@batch2i,i=0,b))

計算兩個張量的點乘(內乘),兩個張量都是1-D向量。

將矩陣mat1和mat2相乘。如果mat1是壹個n×m張量,mat2是壹個m×p張量,則輸出壹個n×p張量out。

將矩陣mat和向量vec相乘。如果mat是n×m張量,vec是m元1維張量,則輸出壹個n元1維張量。

  • 上一篇:請問腎炎患者在日常生活中如何飲食?
  • 下一篇:耐藥肺結核的治療措施
  • copyright 2024偏方大全网