pytorch/test/cpp_extensions/jit_extension2.cpp
Peter Goldsborough 54a4867675
Bring back C++ extension torch.h (#7310)
* Bring back C++ extension torch.h

* Fix python.h include in python_tensor.cpp
2018-05-05 14:06:27 -07:00

7 lines
114 B
C++

#include <torch/torch.h>
using namespace at;
Tensor exp_add(Tensor x, Tensor y) {
return x.exp() + y.exp();
}