pytorch

1
#include <torch/extension.h>
2

3
using namespace at;
4

5
Tensor exp_add(Tensor x, Tensor y) {
6
  return x.exp() + y.exp();
7
}
8