code: def test_masks_changes_2d():
      i = torch.randn([10, 3, 223, 223])
      l = baal.bayesian.consistent_dropout.ConsistentDropout2d()
      _mask_tests(i, l)
  
file_path: baal/tests/bayesian/consistent_dropout_test.py
imports: import warnings
  from copy import deepcopy
  import pytest
  import torch
  import baal.bayesian.consistent_dropout
label_desc: Use image node_id 9 for calling a global function with example usage: test_masks_changes_2d() without return types