code: def test_forward(self) -> None:
      bsz = 3
      dim, size = 2, 10
      n = 33 * bsz
      memory_bank = MemoryBankModule(size=size)
  
      for i in range(0, n, bsz):
          # see if there are any problems when the bank size
          # is no multiple of the batch size
          output = torch.randn(bsz, dim)
          _, _ = memory_bank(output)
  
file_path: lightly/tests/models/modules/test_memory_bank.py
imports: import re
  import unittest
  import pytest
  import torch
  from lightly.models.modules.memory_bank import MemoryBankModule
label_desc: Use image node_id 3 for calling the TestNTXentLoss obj's underlying member method code with example usage: obj.test_forward() without return types