pytorch api文档:torch.max() vs torch.argmax() 对比
pytorch api文档:torch.argmax()返回张量最大值的索引
pytorch api文档:张量的 .numel()方法-统计元素总数
大模型原理:生成文本原理、演示
大模型原理:实现GPT模型
大模型原理:实现TransformerBlock块类(连接注意力层和线性层等)
pytorch api文档:张量的 backward()方法-反向传播自动计算梯度(求导)
pytorch api文档:nn.MSELoss()的backward()方法-反向传播自动计算梯度(求导)
pytorch api文档:nn.MSELoss()均方差损失
pytorch api文档:nn.Module的named_parameters()方法-返回模型中所有可训练的参数