FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
xtensor-python/include/xtensor-python/pyvectorize.hpp at master · xtensor-stack/xtensor-python · GitHub
xtensor-stack
xtensor-python
Repository navigation
Code
Issues
38
(38)
Pull requests
4
(4)
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
xtensor-python
/
include
/
xtensor-python
/
pyvectorize.hpp
Copy path
More file actions
More file actions
Latest commit
History
History
History
63 lines (52 loc) · 2.02 KB
Breadcrumbs
xtensor-python
/
include
/
xtensor-python
/
pyvectorize.hpp
Copy path
File metadata and controls
63 lines (52 loc) · 2.02 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
/*
**************************************************************************
* Copyright (c) Wolf Vollprecht, Johan Mabille and Sylvain Corlay *
* Copyright (c) QuantStack *
* *
* Distributed under the terms of the BSD 3-Clause License. *
* *
* The full license is in the file LICENSE, distributed with this software. *
***************************************************************************
*/
#
ifndef
PY_VECTORIZE_HPP
#
define
PY_VECTORIZE_HPP
#
include
<
type_traits
>
#
include
"
pyarray.hpp
"
#
include
"
xtensor/core/xvectorize.hpp
"
namespace
xt
{
template
<
class
Func
,
class
R
,
class
... Args>
struct
pyvectorizer
{
xvectorizer<Func, R> m_vectorizer;
template
<
class
F
,
class
= std::
enable_if_t
<!std::is_same<std::
decay_t
<F>, pyvectorizer>::value>>
pyvectorizer
(F&& func)
: m_vectorizer(std::forward<F>(func))
{
}
inline
pyarray<R>
operator
()(
const
pyarray<Args>&... args)
const
{
pyarray<R> res =
m_vectorizer
(args...);
return
res;
}
};
/*
*
* @brief Create numpy universal function from scalar function.
*/
template
<
class
R
,
class
... Args>
inline
pyvectorizer<R (*)(Args...), R, Args...>
pyvectorize
(R (*f)(Args...))
{
return
pyvectorizer<
R
(*)(Args...), R, Args...>(f);
}
//
/ @cond DOXYGEN_INCLUDE_OVERLOADS
template
<
class
F
,
class
R
,
class
... Args>
inline
pyvectorizer<F, R, Args...>
pyvectorize
(F&& f, R (*)(Args...))
{
return
pyvectorizer<F, R, Args...>(std::forward<F>(f));
}
template
<
class
F
>
inline
auto
pyvectorize
(F&& f) -> decltype(pyvectorize(std::forward<F>(f), (detail::get_function_type<F>*)nullptr))
{
return
pyvectorize
(std::forward<F>(f), (detail::get_function_type<F>*)
nullptr
);
}
//
/ @endcond
}
#
endif
Back
|
FazBrowse Home
|
New Git URL