AttributeError: 模块 'torch.nn' 没有属性 'linear'

作者: Iwillsd创建于 2021年8月22日更新于 2022年11月23日

=============================================== # 2. 基本自动梯度示例2 # =============================================== # 创建形状为(10,3)和(10,2)的张量 x = torch.randn(10, 3) y = torch.randn(10, 2) # 构建全连接层 linear = nn.linear(3, 2) print('w:', linear.weight) print('b:', linear.bias) # 构建损失函数和优化器 loss = nn.MSELoss() optimizer = torch.optim.SGD(linear.parameters(), lr=0.01)

内容来源: yunjey/pytorch-tutorial