def __init__(self):
super(NetReduceLogic, self).__init__()
self.axis0 = 0
self.axis1 = -1
self.axis2 = (0, 1, 2)
self.axis3 = ()
self.reduce_all = P.ReduceAll(False)
self.reduce_any = P.ReduceAny(False)
def construct(self):
return (P.ReduceAny(self.keep_dims0)(self.x0, self.axis0),
P.ReduceAny(self.keep_dims1)(self.x1, self.axis1),
P.ReduceAny(self.keep_dims2)(self.x2, self.axis2),
P.ReduceAny(self.keep_dims3)(self.x3, self.axis3))
def __init__(self, x, axis):
super(ReduceAnyDynamic, self).__init__()
self.reduceany = P.ReduceAny(False)
self.test_dynamic = inner.GpuConvertToDynamicShape()
self.x = x
self.axis = axis