little speed up by replacing zeros_likes by .fill(0)