pytorch

1
#include <ATen/Tensor.h>
2

3
namespace custom {
4
namespace native {
5
at::Tensor& add_3_out(const at::Tensor& a, const at::Tensor& b, const at::Tensor& c, at::Tensor& out) {
6
    out = a.add(b).add(c);
7
    return out;
8
}
9
}
10
}
11