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