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