How can I change the spatial dimension to the color dimension?(image warping with optical flow in Pytorch)

Question

I’ve some questions of image warping with optical flow.

import torch

from torch.autograd import Variable

import cv2