code: def test_i3d_selayer_shapes(se_layers):
      # test I3D with SELayers
      se_rgb = SEInceptionI3DRGB(3, 8, se_layers)
      se_flow = SEInceptionI3DFlow(2, 8, se_layers)
      se_rgb.eval()
      se_flow.eval()
  
      output_batch = se_rgb(INPUT_BATCH_RGB)
      assert output_batch.size() == (2, 1024, 1)
      output_batch = se_flow(INPUT_BATCH_FLOW)
      assert output_batch.size() == (2, 1024, 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 4 for calling a global function with example usage: test_i3d_selayer_shapes(se_layers) without return types