nestedtensor icon indicating copy to clipboard operation
nestedtensor copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

About gather operator

Open linyangsdu opened this issue 5 years ago • 1 comments

I want to use torch.gather for NestedTensor, but report this error

RuntimeError: Could not run 'aten::gather' with arguments from the 'PrivateUse1' backend.

How can I solve it?

linyangsdu avatar Oct 18 '20 17:10 linyangsdu

Hello @linyangsdu,

gather is currently not implemented. Could you post a snippet with your intended use so we can implement it?

Thanks, Christian

cpuhrsch avatar Oct 21 '20 03:10 cpuhrsch