def forward(self, x):
    x = self.conv(x)
    x = x.reshape(x.size(0), (- 1))
    x = self.linear(x)
    x = self.tanh(x)
    return x
def forward(self, x):
    x = self.conv(x)
    x = x.reshape(x.size(0), (- 1))
    x = self.linear(x)
    x = self.tanh(x)
    return x