def test_Parallel(self):
input = torch.randn(3, 4, 5)
m = nn.Parallel(0, 2)
m.add(nn.View(4, 5, 1))
m.add(nn.View(4, 5, 1))
m.add(nn.View(4, 5, 1))
# Check that these don't raise errors
m.__repr__()
str(m)
output = m.forward(input)
output2 = input.transpose(0, 2).transpose(0, 1)
self.assertEqual(output2, output)
gradInput = m.backward(input, output2)
self.assertEqual(gradInput, input)
评论列表
文章目录