docstr: Returns the distribution of DeepVAR with respect to the RNN outputs.
  
  Parameters
  ----------
  rnn_outputs
      Outputs of the unrolled RNN (batch_size, seq_len, num_cells)
  scale
      Mean scale for each time series (batch_size, 1, target_dim)
  
  Returns
  -------
  distr
      Distribution instance
  distr_args
      Distribution arguments
docstr_tok: ["Returns","the","distribution","of","DeepVAR","with","respect","to","the","RNN","outputs",".","Parameters","--","--","--","--","--","rnn_outputs","Outputs","of","the","unrolled","RNN","(","batch_size",",","seq_len",",","num_cells",")","scale","Mean","scale","for","each","time","series","(","batch_size",",","1",",","target_dim",")","Returns","--","--","--","-","distr","Distribution","instance","distr_args","Distribution","arguments"]
code: def distr_args(self, decoder_output: torch.Tensor):
      """
      Returns the distribution of DeepVAR with respect to the RNN outputs.
  
      Parameters
      ----------
      rnn_outputs
          Outputs of the unrolled RNN (batch_size, seq_len, num_cells)
      scale
          Mean scale for each time series (batch_size, 1, target_dim)
  
      Returns
      -------
      distr
          Distribution instance
      distr_args
          Distribution arguments
      """
      (distr_args,) = self.proj_dist_args(decoder_output)
  
      # # compute likelihood of target given the predicted parameters
      # distr = self.distr_output.distribution(distr_args, scale=scale)
  
      # return distr, distr_args
      return distr_args
  
file_path: gluonts/src/gluonts/nursery/robust-mts-attack/pts/model/transformer_tempflow/transformer_tempflow_network.py
imports: from typing import List, Optional, Tuple
  import torch
  import torch.nn
  from gluonts.core.component import validated
  from pts.modules import RealNVP, MAF, FlowOutput, MeanScaler, NOPScaler
label_desc: Use image node_id 4 for calling the TransformerTempFlowTrainingNetwork obj's underlying member method code with example usage: obj.distr_args(decoder_output) and returns: distr_args