class model(nn.Module):
     ...

     def forward(self, x):
              a = tensor.zeros(1, 256, 64, 48)
              ...

y = model(input).cuda()

if i call model like this then local parameter a is in GPU?