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.

Hi,

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

Thanks,
NVIDIA Enterprise Support