mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01: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
8 lines
115 B
C++
8 lines
115 B
C++
#include <torch/python.h>
|
|
|
|
using namespace at;
|
|
|
|
Tensor exp_add(Tensor x, Tensor y) {
|
|
return x.exp() + y.exp();
|
|
}
|