2024-11-13 厉飞雨 阅读(4) 评论(0) 赞(0)

叶子张量是 PyTorch 计算图中的一个重要概念，叶子张量指的就是我们的模型参数，而模型参数一般都是我们自己创建的 `requires_grad=True` 的张量。它位于整个计算图的开始位置，比如下面这个例子： ``` import torch def test01(): a = torch.tensor([[2.0, 1.0]], requires_grad...