Source code for harl.models.base.flatten

from torch import nn




[docs]
class Flatten(nn.Module):


[docs]
    def forward(self, x):
        return x.view(x.size(0), -1)