def forward(self, x, to_numpy=False, theta=None):
    if (not isinstance(x, torch.cuda.FloatTensor)):
        x = tensor(x)
    if ((x.shape[0] == 1) and (x.shape[1] == 1)):
        x = x[0]
    phi = self.body(x)
    return self.head(phi)
def forward(self, x, to_numpy=False, theta=None):
    if (not isinstance(x, torch.cuda.FloatTensor)):
        x = tensor(x)
    if ((x.shape[0] == 1) and (x.shape[1] == 1)):
        x = x[0]
    phi = self.body(x)
    return self.head(phi)