docs: public void setInitialTensorsToIdentity() {
      for (int i = 0; i < families.size(); i++) {
        LrtFamily lrtFamily = families.get(i);
        if (lrtFamily.getDimensionNumbers().length >= 2) {
  	  Tensor initialValue = SparseTensor.diagonal(lrtFamily.getDimensionNumbers(),
  				lrtFamily.getDimensionSizes(), 1.0);
  	  lrtFamily.setInitialTensor(initialValue);
        }
      }
    }