Hello all,
I am wondering if there is a way to “crop” a tensor with a Relay operator.
So for example (NHWC), I have a tensor with dimensions [2,12,10,4] and I want to only continue processing a subset of it, for example [2,12,10,2] in which the new channels are some specific subset of the previous ones. The remaining cropped-out elements should be deleted/no longer processed.
I see there is a take operator, which I think could do what I am describing but I am unsure.
Thank you very much