當前位置:成語大全網 - 書法字典 - Pytorch火炬級

Pytorch火炬級

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

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

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

賦值公式如下:

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

示例:

根據掩碼張量中的二進制值,取輸入張量中的指定項(掩碼是壹個字節序)並將該值返回到壹個新的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=tensor+value。

另壹個張量的每個元素乘以壹個標量值,並與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)。

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

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

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

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

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

對存儲在批次1和批次2中的矩陣執行批次矩陣乘法。

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

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

對存儲在批次1和批次2中的矩陣執行批次矩陣乘法。batch1和batch2都是包含相同數量矩陣的三維張量。如果batch1是形狀為b×n×m的張量,batch1是形狀為b×m×p的張量,那麽out和mat的形狀都是n×p,即RES =(beta?m)+(阿爾法?sum(batch 1i @ batch 2 I,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維張量。