torch.sparse.as_sparse_gradcheck

torch.sparse.as_sparse_gradcheck(gradcheck)

裝飾器函式，用於擴充套件稀疏張量的 gradcheck。

裝飾器，用於 `torch.autograd.gradcheck` 或其 `functools.partial` 變體，為輸入函式增加了對操作或/和返回稀疏張量的支援。

指定的 `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