By using hybridblock I wrote-
def hybrid_forward(self, F, x):
if F.mean(x) < 1:
return x
else:
return x - 1
While hybridizing it shows error, that bool is not implemented for Symbol.
Is there a way to implement control flow like in above example. Or how to implement a model with data depending forward propagation in gluon.
I know how to use mx.symbol.where but I want to build it using gluon and then hybridize the model.