I have an RCC which outputs tensorrs of size 516x24 that I’d like to get to size 4x6. I have a single linear layer that reduces it to 516x6, but I don’t know how to reduce the 1st dim of 516. Any suggestions? Thanks!