code: def test_dlpack_tensor_gpu_direct_creation():
      arr = torch.rand(size=[3, 5, 6], device="cuda")
      tensor = TensorGPU(to_dlpack(arr))
      dali_torch_tensor = convert_to_torch(
          tensor, device=arr.device, dtype=arr.dtype
      )
      assert torch.all(arr.eq(dali_torch_tensor))
  
imports: from nvidia.dali.tensors import TensorCPU, TensorGPU, TensorListCPU, TensorListGPU
  import nvidia.dali.tensors
  import numpy
  import torch
  from torch.utils.dlpack import to_dlpack
  import ctypes
  from nvidia.dali.backend import CheckDLPackCapsule
label_desc: Use image node_id 2 for calling a global function with example usage: test_dlpack_tensor_gpu_direct_creation() without return types