I’m working on a deep learning problem where the goal is to establish an injective and surjective mapping between elements of an input sequence and elements of an output sequence.
The input sequence consists of feature representations of jigsaw puzzle pieces (extracted by EfficientNetV2L-S, and the output elements are tuples of row, column, and rotation indices in their respective coordinate systems: