torch.sparse.as_sparse_gradcheck

torch.sparse.as_sparse_gradcheck(gradcheck)

用于装饰函数，以扩展对稀疏张量的梯度检查支持。

这是一个用于 torch.autograd.gradcheck 或其 functools.partial 变体的装饰器，它扩展了 gradcheck 函数，使其支持处理或/和返回稀疏张量的输入函数。

所指定的 gradcheck 函数本身保证只处理跨步张量（strided tensors）。

例如

>>> 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