pytorch gathering from a 1d tensor

前端 未结 0 1803
猫巷女王i
猫巷女王i 2021-01-19 23:23

I have a long list of values stored in a 1d tensor from which I like to gather values. Given a 2d tensor where the second dimension denotes a set of indices, I would like to

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