code: def test_i3d_shapes():
      # test InceptionI3D
      i3d = InceptionI3d()
      i3d.eval()
      output_batch = i3d(INPUT_BATCH_RGB)
      assert output_batch.size() == (2, 1024, 1)
  
      # test InceptionI3D.extract_features
      output_batch = i3d.extract_features(INPUT_BATCH_RGB)
      assert output_batch.size() == (2, 1024, 1, 1, 1)
  
file_path: pykale/tests/embed/test_video_i3d_r3d.py
imports: import pytest
  import torch
  from kale.embed.video_i3d import InceptionI3d
  from kale.embed.video_res3d import BasicBlock, BasicStem, Bottleneck, Conv3DSimple, VideoResNet
  from kale.embed.video_se_i3d import SEInceptionI3DFlow, SEInceptionI3DRGB
  from kale.embed.video_se_res3d import se_r3d_18_flow, se_r3d_18_rgb
  from kale.utils.seed import set_seed
label_desc: Use image node_id 1 for calling a global function with example usage: test_i3d_shapes() without return types