torch.sparse.as_sparse_gradcheck

torch.sparse.as_sparse_gradcheck(gradcheck) [source]

Декоратор для 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