torch.gather()函数_开发者

开发者

torch.gather()函数

相关标签：C#JAVA C++objective-c WPF

解析Pytorch中的torch.gather()函数
参数说明以官方说明为例，gather()函数需要三个参数，输入input，维度dim，以及索引index
开发阅读(19)