I wonder how to do it in C++ torchlib.
I think something like this should work:
tensor.index({"...", Slice(None, 1)})
I wonder how to do it in C++ torchlib.
I think something like this should work:
tensor.index({"...", Slice(None, 1)})