This repository has been archived by the owner on Dec 14, 2024. It is now read-only.
Improve the 1D and 2D shift operators. Specifically:
- Enable the 1D shift operator to operate on tensors of ndim >= 2.
- Enable the 2D shift operator to operate on tensors of ndim >= 3. Additionally, remove the height and width arguments to
create_shift2d_opand assume the input data has the shape(..., height, width, dim).