对于深度学习,我们需要计算模型参数的导数。PyTorch 提供了在反向传播时跟踪导数的能力而 NumPy 则没有,这在Pytorch中被称为“Auto Grad”。...我们只需要将列表作为参数传递给函数,我们就有了它的张量形式。...张量的类型是 Double Tensor 而不是默认的 Float Tensor。这对应于 NumPy 的数据类型是float64,如下所示。..., tuple of ints dim, bool keepdim, *, torch.dtype dtype, Tensor out)
* (Tensor input, tuple of names...dim, bool keepdim, *, torch.dtype dtype, Tensor out)
该函数在计算指标和损失函数时非常有用。