1 #pragma once 2 3 #include <ATen/core/Tensor.h> 4 #include <torch/csrc/python_headers.h> 5 6 namespace torch::utils { 7 8 const at::Tensor& apply_(const at::Tensor& self, PyObject* fn); 9 const at::Tensor& map_( 10 const at::Tensor& self, 11 const at::Tensor& other_, 12 PyObject* fn); 13 const at::Tensor& map2_( 14 const at::Tensor& self, 15 const at::Tensor& x_, 16 const at::Tensor& y_, 17 PyObject* fn); 18 19 } // namespace torch::utils 20