code: public SDVariable[] getOutputVariablesForFunction(DifferentialFunction function) {
          val inputs = getOutputsForFunction(function);
          if (inputs == null) {
              throw new ND4JIllegalStateException("No inputs found for function " + function);
          }
  
          val vars = new SDVariable[inputs.length];
          for (int i = 0; i < inputs.length; i++) {
              vars[i] = getVariable(inputs[i]);
          }
  
          return vars;
      }
docs: Get the output variable(s) for the specified differential function
  
  @param function the function reference to get the output variable(s) for
  @return the output variables for the given function