code: def test_r3d_selayer_shapes(se_layers):
      # test R3D with SELayers
      se_rgb = se_r3d_18_rgb(attention=se_layers, pretrained=False)
      se_flow = se_r3d_18_flow(attention=se_layers, pretrained=False)
      se_rgb.eval()
      se_flow.eval()
  
      output_batch = se_rgb(INPUT_BATCH_RGB)
      assert output_batch.size() == (2, 512)
      output_batch = se_flow(INPUT_BATCH_FLOW)
      assert output_batch.size() == (2, 512)
  
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 5 for calling a global function with example usage: test_r3d_selayer_shapes(se_layers) without return types