how to get a col from a n*3 array where n is a variable representing batch size.

if i use gather, the index will be 1 however it need to offer a tensor. mayBe transpose is a choice.


Can you provide some sample code and elaborate on your issue? I don’t quite understand the question.

NVIDIA Enterprise Support