Wondering if anyone knows how to efficiently iterate over a tensor of size 128xAxB? (Where A and B vary).
I’d specifically like to extract the individual values of each index from across dimensions, resulting in a set of 128-element tensors which maintain the order of the original input.
E.g. tensor would extract values in the 0 index across all 128 dimensions, giving a tensor of 128 0-index values. tensor  would extract all values in the index 1 across all 128 dimensions, giving another 128-element tensor, and so on.