I have created a model which inherits nn.Module and I have transferred to model.cuda() . While passing the parameters in forward function i want one argument to be in cpu but since I have used model.cuda() all arguments are in turn being converted to gpu i think. How do i force a tensor to live in cpu ?