docs: public SparseTensor getOldKeyIndicatorTensor() {
      double[] values = new double[oldKeyNums.length];
      Arrays.fill(values, 1.0);
      return SparseTensor.fromUnorderedKeyValues(oldTensor.getDimensionNumbers(), 
          oldTensor.getDimensionSizes(), oldKeyNums, values);
    }