Pytorch - 创建新变量时是否会转移渐变？

Question

我有以下代码：A = Tensor of [186,3]如果我创建一个新的空张量如下：tempTens = torch.tensor(np.zeros((186,3)), requires_grad = True).cuda()我对 A 块应用一些操作并将其输出到 tempTens，我完全使用它进行进一步计算，如下所示：tempTens[20,:] = SomeMatrix * A[20,:]梯度实际上是否会正确传输，假设我有一个成本函数，可以将 tempTens 的输出优化为某些基本事实