Also assert that the output dtype if float64. This is not important as the Mean op is not used by default.
