奋斗的小仔

摘要： 均方误差(Mean Squared Error)是度量模型性能的一种方法。 假设m是样本集的总个数 是第i个样本的预测值，是第i个样本的真实值。 pytorch中的均方误差函数 torch.nn.functional.mse_loss(input, # 预测target, # 目标) 代码实现： > 阅读全文

摘要： 在某些情况下，不需要求出当前张量对所有产生该张量的叶子节点的梯度，这时可以使用torch.autograd.grad()函数。 torch.autograd.grad( outputs, # 计算图的数据结果张量--它就是需要进行求导的函数 inputs, # 需要对计算图求导的张量--它是进行求导 阅读全文