torch.gather理解查看如下代码输出结果:有观察可知,输出的size和index的size是一样大小的。就是说将index放在那,然后对index...,CodeAntenna技术文章技术问题代码 ...
確定! 回上一頁