最近看代码时遇到了两个函数,查阅pytorch官方文档后一时半会儿也没弄懂,现在写篇笔记来加深一下印象。gathertorch.gather(input, dim, index, ...
確定! 回上一頁