def __init__(self):
SparseModule.__init__(self)
self.module = nn.Tanh()
self.output = SparseConvNetTensor()
self.output.features = self.module.output
self.gradInput = self.module.gradInput
def callback(module):
if isinstance(module, nn.ReLU):
return nn.Tanh()
return module
def test_tanh(self):
self._test_single_layer(nn.Tanh())