komunistbakkal t1_izlqyou wrote on December 10, 2022 at 12:57 AM Reply to Does anyone know how to get the NxNx12 from the input image - is it just using reshape function or is there any other function that can be used by Actual-Performer-832 It can also be pixel unshuffle.Here you can find pytorch reference: https://pytorch.org/docs/stable/generated/torch.nn.PixelUnshuffle.html Permalink 2
komunistbakkal t1_izlqyou wrote
Reply to Does anyone know how to get the NxNx12 from the input image - is it just using reshape function or is there any other function that can be used by Actual-Performer-832
It can also be pixel unshuffle.Here you can find pytorch reference: https://pytorch.org/docs/stable/generated/torch.nn.PixelUnshuffle.html