How should I flatten the data before feeding to FC layer , i.e The other way round.
Will this work? Is it necessary to create init method ?
class Flatten(nn.Module):
def forward(self, x):
x = x.view(x.size()[0], -1)
return x