torch.sparse.as_sparse_gradcheck¶
- torch.sparse.as_sparse_gradcheck(gradcheck)[源代码]¶
装饰函数,扩展稀疏张量的 gradcheck。
用于 torch.autograd.gradcheck 或其 functools.partial 变体的装饰器,它扩展了 gradcheck 函数以支持对操作或返回稀疏张量的输入函数的支持。
指定 gradcheck 函数本身保证仅对步幅张量进行操作。
例如
>>> gradcheck = torch.sparse.as_sparse_gradcheck(torch.autograd.gradcheck) >>> x = torch.tensor([[0, 1], [2, 3]], dtype=torch.float64).to_sparse_coo().requires_grad_(True) >>> gradcheck(lambda x: x.to_sparse_csr(), x) True