code: def _post_process_flash_output(out: torch.Tensor, og_size):
      if not out.is_nested and out.size(-1) != og_size:
          out = out[..., 0:og_size]
      return out
  
imports: import logging
  import math
  from typing import Optional, Tuple
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.backends.cuda import can_use_efficient_attention, can_use_flash_attention, flash_sdp_enabled, math_sdp_enabled, mem_efficient_sdp_enabled, SDPAParams, SDPBackend
  from .nested_tensor import buffer_from_jagged, NestedTensor, ViewNestedFromBuffer
label_desc: Use image node_id 17 for calling a global function with example usage: _post_process_flash_output(out, og_size) and returns: out