class_docstr: A structured representation of a policy output over a full (flat) environment step.
class_docstr_tok: ["A","structured","representation","of","a","policy","output","over","a","full","(","flat",")","environment","step","."]
docstr: Get a specified (by index) substep output
docstr_tok: ["Get","a","specified","(","by","index",")","substep","output"]
code: def __getitem__(self, item: int) -> PolicySubStepOutput:
      """Get a specified (by index) substep output"""
      return self._step_policy_outputs[item]
  
file_path: maze/maze/core/agent/torch_policy_output.py
imports: import dataclasses
  from typing import List, Dict, Optional
  import torch
  from maze.core.env.action_conversion import TorchActionType
  from maze.core.env.structured_env import ActorID
  from maze.distributions.dict import DictProbabilityDistribution
label_desc: Use image node_id 2 for calling the PolicyOutput obj's underlying member method code with example usage: obj.__getitem__(item) and returns: self