#pragma once // @generated by torchgen/gen.py from Function.h #include #include #include #include #include #include #include #include #include #include #include #include #include namespace at { // aten::select.Dimname(Tensor(a) self, Dimname dim, int index) -> Tensor(a) inline at::Tensor select(const at::Tensor & self, at::Dimname dim, int64_t index) { return at::_ops::select_Dimname::call(self, dim, index); } // aten::select.int(Tensor(a) self, int dim, SymInt index) -> Tensor(a) inline at::Tensor select(const at::Tensor & self, int64_t dim, int64_t index) { return at::_ops::select_int::call(self, dim, index); } namespace symint { template ::value>> at::Tensor select(const at::Tensor & self, int64_t dim, int64_t index) { return at::_ops::select_int::call(self, dim, index); } } // aten::select.int(Tensor(a) self, int dim, SymInt index) -> Tensor(a) inline at::Tensor select_symint(const at::Tensor & self, int64_t dim, c10::SymInt index) { return at::_ops::select_int::call(self, dim, index); } namespace symint { template ::value>> at::Tensor select(const at::Tensor & self, int64_t dim, c10::SymInt index) { return at::_ops::select_int::call(self, dim, index); } } }