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