class_docstr: Simple feed forward policy network.
docstr: forward pass. 
code: def forward(
      self, in_dict: Dict[str, torch.Tensor]
  ) -> Dict[str, torch.Tensor]:
      """forward pass."""
      return {ACTION_NAME: self.net(in_dict[OBSERVATION_NAME])}
  
file_path: maze/docs/source/policy_and_value_networks/code_snippets/sampling_from_policy.py
imports: from typing import Dict, Sequence
  import torch
  from gym import spaces
  from torch import nn
  from maze.distributions.distribution_mapper import DistributionMapper
label_desc: Use image node_id 2 for calling the PolicyNet obj's underlying member method code with example usage: obj.forward(in_dict) and returns: dict