- 博客(3)
- 收藏
- 关注
RSS订阅翻译 繁星*春水
注意如果中间张量，既不是forward（）的输入，也不是输出张量，被保存在backward（）中，你的自定义函数不支持double backward()。所有的tensors 计划被用来向后传播应该使用save_for_backward(而不是直接在ctx）保存，来避免不正确的梯度和内存泄漏.并且启用torch.autograd.graph.saved_tensors_hooks。save_for_backward()建议最多被调用一次，只能从forward()方法内部调用，并且只能使用Tensor张量。
2024-03-23 14:47:44
53
原创 关于tensorflow学习（关于如何构建网络来计算偏微分方程）。
后面会发具体的流程，想要文章源码的私我哦原文链接：https://blog.csdn.net/huang1024rui/article/details/120055487。
2024-03-13 17:18:16
322
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人