mirror of
https://github.com/boostorg/python.git
synced 2026-01-31 08:22:18 +00:00
Merge NumPy extension from https://github.com/ndarray/Boost.NumPy/.
This commit is contained in:
21
test/numpy/shapes_mod.cpp
Normal file
21
test/numpy/shapes_mod.cpp
Normal file
@@ -0,0 +1,21 @@
|
||||
// Copyright Jim Bosch & Ankit Daftery 2010-2012.
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
|
||||
#include <boost/numpy.hpp>
|
||||
|
||||
namespace p = boost::python;
|
||||
namespace np = boost::numpy;
|
||||
|
||||
np::ndarray reshape(np::ndarray old_array, p::tuple shape)
|
||||
{
|
||||
np::ndarray local_shape = old_array.reshape(shape);
|
||||
return local_shape;
|
||||
}
|
||||
|
||||
BOOST_PYTHON_MODULE(shapes_mod)
|
||||
{
|
||||
np::initialize();
|
||||
p::def("reshape", reshape);
|
||||
}
|
||||
Reference in New Issue
Block a user