How to use torch.gather

前端 未结 0 804
北恋
北恋 2021-02-04 11:17

I have a tensor of size torch.Size([3307, 1, 3]) containing values and another tensor of size torch.Size([3307, 1, 1]) containing indices of values I want to select. I am unable

相关标签:
回答
  • 消灭零回复
提交回复
热议问题