code: def move_batch(self, batch: TorchTuple, non_blocking=False):
      return batch.to(self.device, non_blocking)
  
imports: import torch
  import torch.nn
  from torch.nn import CrossEntropyLoss, MSELoss
  from turing.utils import TorchTuple
  from pytorch_pretrained_bert.modeling import BertModel
  from pytorch_pretrained_bert.modeling import BertPreTrainingHeads, PreTrainedBertModel, BertPreTrainingHeads
  from pytorch_pretrained_bert.file_utils import PYTORCH_PRETRAINED_BERT_CACHE
label_desc: Use image node_id 5 for calling the BertMultiTask obj's underlying member method code with example usage: obj.move_batch(batch, non_blocking) and returns: batch