/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils/tensor_apply.h (431B)
#pragma once
#include
#include
namespace torch { namespace utils {
const at::Tensor & apply_(const at::Tensor & self, PyObject* fn);
const at::Tensor & map_(const at::Tensor & self, const at::Tensor & other_, PyObject* fn);
const at::Tensor & map2_(const at::Tensor & self, const at::Tensor & x_,
const at::Tensor & y_, PyObject* fn);
}} // namespace torch::utils