改進之處包括:使torch.子模塊導入更易於自動完成、在ONNX 中添加對torch.{isinf,any,all}的支持、在torch.randperm的GPU 實現中將thrust 替換為cub, ...
確定! 回上一頁