📌  相关文章
📜  火炬均方根 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:12.473000             🧑  作者: Mango

代码示例1
#its not built in but you can get RMSE like so:
criterion = nn.MSELoss()
loss = torch.sqrt(criterion(x, y))
loss.backward()
print(x.grad)