why do you fix it here? Why not fix it in compyte? I think compyte should support that. This would mean that compyte elemwise don't support shape of 0.