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