Ejemplo n.º 1
0
 def forward_chainerx(self, x):
     return chainerx.amax(x[0], axis=self.axis, keepdims=self.keepdims),
Ejemplo n.º 2
0
 def forward_chainerx(self, x):
     return chainerx.amax(x[0], axis=self.axis, keepdims=self.keepdims),