code: def test_pt_autoregressive_mask(length):
      x_pt = pt.zeros(2, length, 32)
      b_pt = sockeye.transformer.AutoRegressiveMask()
      result_pt = b_pt(x_pt).detach()
  
      assert result_pt.dtype == pt.bool
      assert result_pt.size() == (length, length)
  
imports: import pytest
  import torch
  import sockeye.constants
  import sockeye.transformer
label_desc: Use image node_id 2 for calling a global function with example usage: test_pt_autoregressive_mask(length) without return types