Also assert that the output dtype if float64.

This is not important as the Mean op is not used by default.