docstr: Get the slate_size for averaging the sum of slate value.
docstr_tok: ["Get","the","slate_size","for","averaging","the","sum","of","slate","value","."]
code: def _get_avg_by_slate_size(self, batch: rlt.SlateQInput):
      """Get the slate_size for averaging the sum of slate value."""
      if (
          self.next_slate_value_norm_method
          == NextSlateValueNormMethod.NORM_BY_NEXT_SLATE_SIZE
      ):
          return self._get_slate_size(batch.next_state)
      if (
          self.next_slate_value_norm_method
          == NextSlateValueNormMethod.NORM_BY_CURRENT_SLATE_SIZE
      ):
          return self._get_slate_size(batch.state)
      raise NotImplementedError(
          f"The next_slate_value_norm_method {self.next_slate_value_norm_method} has not been implemented"
      )
  
imports: import enum
  import logging
  from typing import Optional
  import reagent.core.parameters
  import reagent.core.types
  import torch
  import torch.nn.functional
  from reagent.core.dataclasses import field
  from reagent.optimizer import Optimizer__Union, SoftUpdate
  from reagent.training.reagent_lightning_module import ReAgentLightningModule
  from reagent.training.rl_trainer_pytorch import RLTrainerMixin
label_desc: Use image node_id 10 for calling the SlateQTrainer obj's underlying member method code with example usage: obj._get_avg_by_slate_size(batch) and returns: self, self