code: def test_get_used_tokens_for_gbm():
      inputs = {
          "input1": torch.tensor([[1, 2], [3, 4]]),
          "input2": torch.tensor([5, 6]),
      }
  
      assert get_used_tokens_for_gbm(inputs) == 3
  
imports: import torch
  from ludwig.accounting.used_tokens import get_used_tokens_for_ecd, get_used_tokens_for_gbm, get_used_tokens_for_llm
label_desc: Use image node_id 1 for calling a global function with example usage: test_get_used_tokens_for_gbm() without return types