mirror of
https://github.com/boostorg/ublas.git
synced 2026-01-19 04:42:15 +00:00
231ba5f730
Auxiliary functions for extents and strides were using different functions. Additionally, many tags were used to distinguish between different tensor types. This patch simplifies interfaces of different core functions and unifies functions that can process different types of extent and stride types.
117 lines
4.1 KiB
C++
117 lines
4.1 KiB
C++
//
|
|
// Copyright (c) 2018, Cem Bassoy, cem.bassoy@gmail.com
|
|
// Copyright (c) 2019, Amit Singh, amitsingh19975@gmail.com
|
|
//
|
|
// 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)
|
|
//
|
|
// The authors gratefully acknowledge the support of
|
|
// Google and Fraunhofer IOSB, Ettlingen, Germany
|
|
//
|
|
|
|
#include <boost/numeric/ublas/tensor.hpp>
|
|
#include <boost/multiprecision/cpp_bin_float.hpp>
|
|
|
|
#include <ostream>
|
|
|
|
//NOLINTNEXTLINE
|
|
int main()
|
|
{
|
|
namespace ublas = boost::numeric::ublas;
|
|
|
|
try {
|
|
using value = float;
|
|
using layout = ublas::layout::first_order; // storage format
|
|
using tensor = ublas::tensor_dynamic<value,layout>;
|
|
// constexpr auto ones = ublas::ones<value,layout>{};
|
|
constexpr auto zeros = ublas::zeros<value,layout>{};
|
|
|
|
// creates a three-dimensional tensor with extents 3,4 and 2
|
|
// tensor A stores single-precision floating-point number according
|
|
// to the first-order storage format
|
|
|
|
tensor A = zeros(3,4,2);
|
|
|
|
// initializes the tensor with increasing values along the first-index
|
|
// using a single index.
|
|
auto vf = 1.0f;
|
|
for(auto i = 0u; i < A.size(); ++i, vf += 1.0f)
|
|
A[i] = vf;
|
|
|
|
// formatted output
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "A=" << A << ";" << std::endl << std::endl;
|
|
} catch (const std::exception& e) {
|
|
std::cerr << "Cought exception " << e.what();
|
|
std::cerr << "in the main function of access-tensor." << std::endl;
|
|
}
|
|
|
|
|
|
try {
|
|
using value = std::complex<boost::multiprecision::cpp_bin_float_double_extended>;
|
|
using layout = ublas::layout::last_order; // storage format
|
|
using tensor = ublas::tensor_dynamic<value,layout>;
|
|
using shape = typename tensor::extents_type;
|
|
constexpr auto zeros = ublas::zeros<value,layout>{};
|
|
|
|
|
|
// creates a four-dimensional tensor with extents 5,4,3 and 2
|
|
// tensor A stores complex floating-point extended double precision numbers
|
|
// according to the last-order storage format
|
|
// and initializes it with the default value.
|
|
|
|
//NOLINTNEXTLINE
|
|
tensor B = zeros(5,4,3,2);
|
|
|
|
// initializes the tensor with increasing values along the last-index
|
|
// using a single-index
|
|
auto vc = value(0,0);
|
|
for(auto i = 0u; i < B.size(); ++i, vc += value(1,1))
|
|
B[i] = vc;
|
|
|
|
// formatted output
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "B=" << B << ";" << std::endl << std::endl;
|
|
|
|
|
|
auto C = tensor(B.extents());
|
|
// computes the complex conjugate of elements of B
|
|
// using multi-index notation.
|
|
for(auto i = 0u; i < B.size(0); ++i)
|
|
for(auto j = 0u; j < B.size(1); ++j)
|
|
for(auto k = 0u; k < B.size(2); ++k)
|
|
for(auto l = 0u; l < B.size(3); ++l)
|
|
C.at(i,j,k,l) = std::conj(B.at(i,j,k,l));
|
|
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "C=" << C << ";" << std::endl << std::endl;
|
|
|
|
|
|
|
|
// computes the complex conjugate of elements of B
|
|
// using iterators.
|
|
auto D = tensor(B.extents());
|
|
std::transform(B.begin(), B.end(), D.begin(), [](auto const& b){ return std::conj(b); });
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "D=" << D << ";" << std::endl << std::endl;
|
|
|
|
// reshaping tensors.
|
|
auto new_extents = B.extents().base();
|
|
std::next_permutation( new_extents.begin(), new_extents.end() );
|
|
auto E = reshape( D, shape(new_extents) );
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "E=" << E << ";" << std::endl << std::endl;
|
|
|
|
|
|
} catch (const std::exception& e) {
|
|
std::cerr << "Cought exception " << e.what();
|
|
std::cerr << "in the main function of access-tensor." << std::endl;
|
|
}
|
|
}
|