docs: public void initializeParametersToIdentity(SufficientStatistics parameters) {
      CvsmSufficientStatistics cvsmStats = (CvsmSufficientStatistics) parameters;
      Preconditions.checkArgument(cvsmStats.size() == families.size());
  
      for (int i = 0; i < cvsmStats.size(); i++) {
        SufficientStatistics curStats = cvsmStats.getSufficientStatistics(i);
        LrtFamily lrtFamily = families.get(i);
        if (lrtFamily.getDimensionNumbers().length >= 2) {
          if (curStats instanceof TensorSufficientStatistics) {
            TensorSufficientStatistics tensorStats = (TensorSufficientStatistics) curStats;
            Tensor tensor = tensorStats.get();
            Tensor diag = SparseTensor.diagonal(tensor.getDimensionNumbers(),
                tensor.getDimensionSizes(), 1.0);
            tensorStats.increment(diag, 1.0);
          } else {
            List<SufficientStatistics> stats = curStats.coerceToList().getStatistics();
            if (stats.size() == 0) {
              // ConstantLrtFamily.
              continue;
            }
            TensorSufficientStatistics diagStats = (TensorSufficientStatistics) stats.get(stats.size() - 1);
            Tensor tensor = diagStats.get();
  
            DenseTensor increment = DenseTensor.constant(tensor.getDimensionNumbers(), tensor.getDimensionSizes(), 1.0);
            diagStats.increment(increment, 1.0);
          }
        }
      }
    }