class_docstr: Quantile action value for discrete actions.
  Args:
      quantiles (torch.Tensor): (batch_size, n_taus, n_actions)
      q_values_formatter (callable):
class_docstr_tok: ["Quantile","action","value","for","discrete","actions",".","Args",":","quantiles","(","torch.Tensor",")",":","(","batch_size",",","n_taus",",","n_actions",")","q_values_formatter","(","callable",")",":"]
code: def __init__(self, quantiles, q_values_formatter=lambda x: x):
      assert quantiles.ndim == 3
      self.quantiles = quantiles
      self.n_actions = quantiles.shape[2]
      self.q_values_formatter = q_values_formatter
  
imports: import warnings
  from abc import ABCMeta, abstractmethod, abstractproperty
  import torch
  import torch.nn.functional
  from torch.distributions.utils import lazy_property
label_desc: Use image node_id 1 to create a new QuantileDiscreteActionValue object from inherited base classes: DiscreteActionValue with example: obj = QuantileDiscreteActionValue(quantiles, q_values_formatter)