mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
67d0d14908
* Rename autograd namespace to torch and change torch.h into python.h * Include torch.h instead of python.h in test/cpp/api * Change some mentions of torch.h to python.h in C++ extensions * Set paths directly, without find_path
7 lines
115 B
C++
7 lines
115 B
C++
#include <torch/python.h>
|
|
|
|
using namespace at;
|
|
|
|
Tensor exp_add(Tensor x, Tensor y) {
|
|
return x.exp() + y.exp();
|
|
}
|