code: def plate_cuda_model(subsample_size):
      loc = torch.zeros(20).cuda()
      scale = torch.ones(20).cuda()
      with pyro.plate(
          "data", 20, subsample_size, device=loc.device
      ) as batch:
          pyro.sample("x", dist.Normal(loc[batch], scale[batch]))
  
file_path: pyro/tests/poutine/test_mapdata.py
imports: import logging
  import pytest
  import torch
  import pyro
  import pyro.distributions
  import pyro.poutine
  from tests.common import requires_cuda
label_desc: Use image node_id 10 for calling a global function with example usage: plate_cuda_model(subsample_size) without return types