欢迎光临散文网 会员登陆 & 注册

尹成人工智能Pytorch1.9

2023-04-28 11:11 作者:s陌上烟花  | 我要投稿

libtorch新算子导出.so,python调用

index_add.cpp

#include <torch/extension.h>                    // 头文件引用部分 torch::Tensor index_add(torch::Tensor &left, torch::Tensor &index, torch::Tensor &right) {        for (int i = 0; i < index.size(0); i++){        left[index[i]] = left[index[i]] + right[i];    }



尹成人工智能Pytorch1.9的评论 (共 条)

分享到微博请遵守国家法律