PyTorch的linear函數用于定義一個線性變換。它可以將輸入張量與權重矩陣進行矩陣乘法,并加上偏置向量,從而得到輸出張量。
具體來說,linear函數的作用是將輸入張量的每個元素與權重矩陣相乘,然后將所有乘積相加,并加上偏置向量的對應元素。這個過程可以表示為以下公式:
output = input * weight^T + bias
其中,input是輸入張量,weight是權重矩陣,bias是偏置向量。權重矩陣的大小為[out_features, in_features],其中out_features表示輸出張量的大小,in_features表示輸入張量的大小。偏置向量的大小為[out_features],即與輸出張量的大小相同。
linear函數可以用于構建神經網絡的全連接層,它可以將輸入張量映射到輸出張量的不同維度,實現特征的轉換和提取。