def assign_tensor(tensor, val): """ copy val to tensor Args: tensor: an n-dimensional torch.Tensor val: an n-dimensional torch.Tensor to fill the tensor with Returns: """ #if isinstance(tensor, Variable): # assign_tensor(tensor.data, val) # return tensor return tensor.data.copy_(val)