code: public Map<String,Field> getFieldsForFunction(DifferentialFunction function) {
          return fieldsForFunction.get(function.opName());
      }
docs: Get the fields for a given {@link DifferentialFunction}
  @param function the function to get the fields for
  @return the fields for a given function