예제 #1
0
 def forward_chainerx(self, x):
     return chainerx.amax(x[0], axis=self.axis, keepdims=self.keepdims),
예제 #2
0
파일: minmax.py 프로젝트: hvy/chainer
 def forward_chainerx(self, x):
     return chainerx.amax(x[0], axis=self.axis, keepdims=self.keepdims),