code: def _test_alibi():
      """
      Simple test function to see the slopes.
      """
      inspect(get_slopes(12).tolist(), _n=-1)
      from labml_nn.transformers.utils import subsequent_mask
  
      mask = subsequent_mask(8)[:, :, 0]
      inspect(mask)
  
      inspect(get_alibi_biases(12, mask)[:, :, 3], _n=-1)
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/transformers/alibi/__init__.py
imports: import math
  from typing import Optional
  import torch
  from torch import nn
  from labml.logger import inspect
  from labml_nn.transformers.mha import MultiHeadAttention
label_desc: Use image node_id 3 for calling a global function with example usage: _test_alibi() without return types