为您找到相关结果621,155个
PyTorch一小时掌握之autograd机制篇_python_脚本之家
y_train = y_train.reshape(-1,1) print(y_train.shape) # (11, 1) # 构造模型 class LinerRegressionModel(nn.Module): def __init__(self, input_dim, output_dim): super(LinerRegressionModel, self).__init__() self.liner = nn.Linear(input_dim, output_dim) def forward(self, x): ou...
www.jb51.net/article/2221...htm 2025-2-14