test_nn.py 文件源码

def test_conv_tbc(self):
        inp = Variable(torch.randn(9, 4, 5), requires_grad=True)
        weight = Variable(torch.randn(3, 5, 6), requires_grad=True)
        bias = Variable(torch.randn(6), requires_grad=True)

        gradcheck(lambda i, w, b, pad: F.conv_tbc(i, w, b, pad), (inp, weight, bias, 3))