The line below should use ```np.zeros_like``` so that it will preserve float32 if that is the input.