yes, call .metal() on your input tensor and call .cpu() on the output tensor to move it back. Something like
auto input = torch.rand({1,3,2,2}).metal()
auto output = model.forward({input}).toTensor().cpu();