PyTorch筆記 torch.nn.Linear

官方文檔:LINEAR 注意到這個公式: 結合代碼講解其中的含義: x的維度:4*2 L的weight的維度:3*2 L的bias的維度:3 y的維度:4*3 根據線性代數的矩陣乘法的知識,可以猜想,是不是 y = x*weight^T + bias(其中,+bias爲廣播)? 驗證一下: 果然,L.weight是公式中的A,L.bias是公式中的b。
相關文章
相關標籤/搜索