Description
🐛 Describe the bug
#61643 broke the originally working case non_fp_tensor.mean(dtype=fp_dtype).
Original code checks
ScalarType scalarType = opt_dtype.has_value() ? opt_dtype.value() : self.scalar_type();
but the new code doesn't consider opt_dtype in checking:
at::isFloatingType(self_dtype) || at::isComplexType(self_dtype),
CC @ysiraichi who authored the PR, and @ezyang who merged it.
Versions
1.10+