fix docstring

Signed-off-by: 0oshowero0 <o0shower0o@outlook.com>

Author: 0oshowero0

transfer_queue/utils/tensor_utils.py

@@ -110,6 +110,7 @@ def compute_stride(shape: tuple[int, ...]) -> tuple[int, ...]:
 
 
 def get_nbytes(dtypes, shapes) -> list[int]:
+    """Calculate number of bytes according to tensor dtypes and shapes."""
     assert len(dtypes) == len(shapes)
     nbytes = []
     for i in range(len(dtypes)):