code: def raggedness_matches(nt, size):
      end = nt._ragged_idx + 1
      nt_ragged = nt._size[:end]
      size_ragged = size[:end]
      return len(nt_ragged) == len(size_ragged) and (
          all(
              ns == s or s == -1
              for ns, s in zip(nt_ragged, size_ragged)
          )
      )
  
imports: import functools
  import math
  import operator
  import torch
  from torch.nested._internal.sdpa import jagged_scaled_dot_product_attention
  from .nested_tensor import NestedTensor
  from typing import *
  from torch.fx.operator_schemas import normalize_function
label_desc: Use image node_id 6 for calling a global function with example usage: raggedness_matches(nt, size) and returns: unknown