code: def __init__(self, action_space: ActionSpace, model):
      self._action_space = action_space
      self._model = model
      self._softmax = torch.nn.Softmax(dim=0)
  
imports: import logging
  import gym
  import torch
  from reagent.ope.estimators.sequential_estimators import Action, ActionDistribution, ActionSpace, IPSEstimator, Model, NeuralDualDICE, RandomRLPolicy, RewardProbability, RLEstimatorInput, RLPolicy, State, StateDistribution, Transition
  from reagent.ope.utils import RunningAverage
label_desc: Use image node_id 1 to create a new PyTorchPolicy object from inherited base classes: RLPolicy with example: obj = PyTorchPolicy(action_space, model)