def METHOD_NAME(self):
    tensors = [torch.ones([2, 2]), torch.ones([1, 3])]
    max_lens = [4, 2]
    expected_output = torch.tensor(
        [[[1, 1, 0, 0], [1, 1, 0, 0]], [[1, 1, 1, 0], [0, 0, 0, 0]]], dtype=torch.float32
    )
    stacked_tensor = stack_tensors(tensors=tensors, max_lens=max_lens)
    torch.testing.assert_close(stacked_tensor, expected_output)