Split with n splits=1 should be optimized away
Split with n splits=1 should be optimized away: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). When I use theano.tensor.split with n splits=1 (this comes up in generic code where the number of splits is computed from a parameter that can be 1), the resulting graph…
The task
When I use `theano.tensor.split` with `n_splits=1` (this comes up in generic code where the number of splits is computed from a parameter that can be 1), the resulting graph still contains a full `Split` Apply node, even though semantically it's a no-op — it just returns the input tensor unchanged along the requested…
Part of FineEnvs/MiMo-V2.6-RL-harbor-code.