docs: public Tensor getFeatureWeightsForClass(Assignment outputClass) {
      int[] classIndexes = getOutputVariables().assignmentToIntArray(outputClass);
      int[] dimensionNums = getOutputVariables().getVariableNumsArray();
  
      return logWeights.slice(dimensionNums, classIndexes);
    }