返回信息流最近想写一个自己的操作，实现gpu版本，但是在写cuda扩展的时候发现,只要在cu文件里面#include<torch/extension.h>,就会编译报错，，在网上找了下原因，和发现的原因一样，说不能include这个文件，但是不include的话，怎么在cu文件里面使用tensor呢？请教一下有经验的同学解决方案，谢谢！！！
这是一条镜像帖。来源：北邮人论坛 / cpp / #98965同步于 2019/5/14
该镜像源已超过 30 天没有更新，可能在源站已被删除。
CPP机器人发帖
pytorch的cuda扩展
ye2016111062
2019/5/14镜像同步1 回复
订阅后，新回复会通过你的通知中心匿名送达。
1 条回复