PyTorch涓殑Linear鍑芥暟鐢ㄤ簬瀹氫箟绾挎€у眰锛屽彲浠ュ皢杈撳叆鏁版嵁鐨勫ぇ灏忔槧灏勫埌杈撳嚭鏁版嵁鐨勫ぇ灏忋€傚畠鏄疨yTorch涓殑涓€涓缁忕綉缁滄ā鍧楋紝鍙互閫氳繃瀹炰緥鍖?code>torch.nn.Linear绫绘潵浣跨敤銆?/p>

浠ヤ笅鏄竴涓娇鐢?code>Linear鍑芥暟鐨勭ず渚嬶細

import torch
import torch.nn as nn

# 瀹氫箟杈撳叆鏁版嵁鐨勫ぇ灏忓拰杈撳嚭鏁版嵁鐨勫ぇ灏?/span>
input_size = 10
output_size = 5

# 瀹炰緥鍖朙inear鍑芥暟
linear_layer = nn.Linear(input_size, output_size)

# 鐢熸垚闅忔満杈撳叆鏁版嵁
input_data = torch.randn(1, input_size)

# 浣跨敤Linear鍑芥暟杩涜鍓嶅悜浼犳挱
output_data = linear_layer(input_data)

print(output_data)

鍦ㄤ笂杩扮ず渚嬩腑锛屾垜浠鍏堝畾涔変簡杈撳叆鏁版嵁鐨勫ぇ灏忎负10锛岃緭鍑烘暟鎹殑澶у皬涓?銆傜劧鍚庡疄渚嬪寲浜嗕竴涓?code>Linear鍑芥暟瀵硅薄linear_layer锛岃瀵硅薄灏嗚緭鍏ユ暟鎹殑澶у皬鏄犲皠鍒拌緭鍑烘暟鎹殑澶у皬銆傛帴涓嬫潵锛屾垜浠敓鎴愪簡涓€涓殢鏈虹殑1x10澶у皬鐨勮緭鍏ユ暟鎹?code>input_data锛屽苟閫氳繃璋冪敤linear_layer瀵硅薄杩涜鍓嶅悜浼犳挱锛屽緱鍒颁簡杈撳嚭鏁版嵁output_data銆?/p>

姝ゅ锛?code>Linear鍑芥暟杩樻湁涓€浜涘叾浠栧彲閫夊弬鏁帮紝渚嬪鏄惁浣跨敤鍋忕疆椤癸紙bias锛夌瓑锛屽彲浠ラ€氳繃淇敼瀹炰緥鍖?code>nn.Linear绫绘椂鐨勫弬鏁版潵璁剧疆杩欎簺閫夐」銆傚叿浣撳彲鍙傝€働yTorch瀹樻柟鏂囨。涓叧浜?code>Linear鍑芥暟鐨勮鏄庛€?/p>