code: def test_spatial_dropout():
      dropout = SpatialDropout()
      x = torch.randn([5, 20, 40], requires_grad=True)
      y = dropout(x)
      assert x.shape == y.shape
  
imports: import torch
  from espnet2.asr.preencoder.sinc import LightweightSincConvs, SpatialDropout
label_desc: Use image node_id 1 for calling a global function with example usage: test_spatial_dropout() without return types