Applying 3D flow on 3D voxel data


Are there any ways to apply 3D flow (displacement) on 3D voxel data?

For example, I have a voxel data (128x128x128) which contains binary values (when there is an object in a cell then contains 1 otherwise 0).
Also, I have a 3D flow (3x128x128x128). I wanna apply the 3D flow on the voxel data.

Do you know any pytorch library for this operation?

Thanks in advance!