code: def get_all_q_values_for_states_and_softmax_probabilities(
      self, states: StateType
  ):
      actions_q_values, softmax_probabilities = None, None
      if self.exploration_policy.requires_action_values():
          outputs = copy(self.networks["main"].online_network.outputs)
          outputs.append(
              self.networks["main"]
              .online_network.output_heads[0]
              .softmax
          )
          quantile_values, softmax_probabilities = self.get_prediction(
              states, outputs
          )
          actions_q_values = self.get_q_values(quantile_values)
  
      return actions_q_values, softmax_probabilities
  
imports: from copy import copy
  from typing import Union
  import numpy
  from rl_coach.agents.dqn_agent import DQNAgentParameters, DQNNetworkParameters, DQNAlgorithmParameters
  from rl_coach.agents.value_optimization_agent import ValueOptimizationAgent
  from rl_coach.architectures.head_parameters import QuantileRegressionQHeadParameters
  from rl_coach.core_types import StateType
  from rl_coach.schedules import LinearSchedule
label_desc: Use image node_id 5 for calling the QuantileRegressionDQNAgent obj's underlying member method code with example usage: obj.get_all_q_values_for_states_and_softmax_probabilities(states) and returns: actions_q_values,  softmax_probabilities