%`i ddlmZddlZddlmcmZddlmZddedededee fd Z dd e d e d e defdZ ddeede fdZ dS))OptionalN)Tensorbilinearimg absolute_gridmode align_cornersc|jdd\}}|ddgd\}}d|z|dz z dz }|dkrd|z|dz z dz }tj||gd}t j||||S)z_Same as torch's grid_sample, with absolute pixel coordinates instead of normalized coordinates.Ndim)rr )shapesplittorchcatF grid_sample) rrrr hwxgridygridnormalized_grids z/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchvision/models/optical_flow/_utils.pyrrs 9RSS>DAq &&1v2&66LE5 IQ ! #E1uuE QU#a'iB777O =oD V V VVcpu batch_sizerrdevicecFtj|}tjtj||tj||d}tj|dddd}|d|dddS)N)r ij)indexingr rrr )rr meshgridarangestackfloatrepeat)rrrr coordss rmake_coords_gridr*s \& ! !F ^EL6:::ELSYrKs  W WV WF W# W[cdh[i W W W W44444c4444dd&!1d#ddddddr