class_docstr: Selection Policy Network for cutting 2d.
  
  :param obs_shapes: The shapes of all observations as a dict.
  :param action_logits_shapes: The shapes of all actions as a dict structure.
  :param non_lin: The nonlinear activation to be used.
  :param with_mask: Weather to use action masking or not.
class_docstr_tok: ["Selection","Policy","Network","for","cutting","2d",".",":","param","obs_shapes",":","The","shapes","of","all","observations","as","a","dict",".",":","param","action_logits_shapes",":","The","shapes","of","all","actions","as","a","dict","structure",".",":","param","non_lin",":","The","nonlinear","activation","to","be","used",".",":","param","with_mask",":","Weather","to","use","action","masking","or","not","."]
docstr: Compute forward pass through the network.
  
  :param xx: Input dict.
  :return: The computed output of the network.
docstr_tok: ["Compute","forward","pass","through","the","network",".",":","param","xx",":","Input","dict",".",":","return",":","The","computed","output","of","the","network","."]
code: def forward(
      self, xx: Dict[str, torch.Tensor]
  ) -> Dict[str, torch.Tensor]:
      """Compute forward pass through the network.
  
      :param xx: Input dict.
      :return: The computed output of the network.
      """
      return self.perception_net(xx)
  
imports: from collections import OrderedDict
  from typing import Dict, Union, Sequence
  import torch
  from maze.perception.blocks.feed_forward.dense import DenseBlock
  from maze.perception.blocks.general.action_masking import ActionMaskingBlock
  from maze.perception.blocks.general.concat import ConcatenationBlock
  from maze.perception.blocks.general.correlation import CorrelationBlock
  from maze.perception.blocks.general.functional import FunctionalBlock
  from maze.perception.blocks.inference import InferenceBlock
  from maze.perception.blocks.output.linear import LinearOutputBlock
  from maze.perception.weight_init import make_module_init_normc
  from torch import nn
label_desc: Use image node_id 2 for calling the SelectionPolicyNet obj's underlying member method code with example usage: obj.forward(xx) and returns: self