Use DenseIndex in the MeanReducer to avoid overflows when processing very large tensors.
1 file changed