Method fn

test/distributed/test_data_parallel.py:47–48

(t)

        inp = torch.randn(2, 100, device='cuda', dtype=torch.double)

        def fn(t):
            return dpm(inp)

        gradcheck(fn, (m.t_rg,))