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。
相关文章
相关标签/搜索