輸入被拼接,參數指示要拼接的尺寸。
沿給定軸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維張量。