flax_1d 0 0 0