改进之处包括:使torch.子模块导入更易于自动完成、在ONNX 中添加对torch.{isinf,any,all}的支持、在torch.randperm的GPU 实现中将thrust 替换为cub, ...
確定! 回上一頁