Example #1
0
 def forward(self, x):
     return LF.gem(x, p=self.p.unsqueeze(-1).unsqueeze(-1), eps=self.eps)
Example #2
0
 def forward(self, x):
     return LF.gem(x, p=self.p, eps=self.eps)