extra ops.repeat gives wrong shape when repeats is a vector and axis=None
extra ops.repeat gives wrong shape when repeats is a vector and axis=None: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). extra ops.repeat gives wrong shape when repeats is a vector and axis=None I'm using theano.tensor.extra ops.repeat to do the same thing as numpy.repeat…
The task
extra_ops.repeat gives wrong shape when repeats is a vector and axis=None I'm using theano.tensor.extra_ops.repeat to do the same thing as numpy.repeat — repeat each element of an array a (possibly different) number of times. When repeats is a 1D tensor and I don't pass axis…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.