mirror of
https://github.com/boostorg/ublas.git
synced 2026-01-19 16:52:13 +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.
74 lines
2.1 KiB
C++
74 lines
2.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/numeric/ublas/matrix.hpp>
|
|
#include <boost/numeric/ublas/vector.hpp>
|
|
#include <ostream>
|
|
|
|
int main()
|
|
{
|
|
namespace ublas = boost::numeric::ublas;
|
|
using value = float;
|
|
using tensor = ublas::tensor_dynamic<value>;
|
|
using matrix = ublas::matrix<value>;
|
|
using vector = ublas::vector<value>;
|
|
using shape = tensor::extents_type;
|
|
|
|
try {
|
|
|
|
|
|
auto A = tensor{3,4,2};
|
|
auto B = A = 2;
|
|
|
|
// Calling overloaded operators
|
|
// and using simple tensor expression templates.
|
|
if( A != (B+1) ){
|
|
A += 2*B - 1;
|
|
}
|
|
|
|
// formatted output
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "A=" << A << ";" << std::endl << std::endl;
|
|
|
|
auto n = shape{3,4};
|
|
auto D = matrix(n[0],n[1],1);
|
|
auto e = vector(n[1],1);
|
|
auto f = vector(n[0],2);
|
|
|
|
// Calling constructor with
|
|
// vector expression templates
|
|
tensor C = 2*f;
|
|
// formatted output
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "C=" << C << ";" << std::endl << std::endl;
|
|
|
|
|
|
// Calling overloaded operators
|
|
// and mixing simple tensor and matrix expression templates
|
|
tensor F = 3*C + 4*prod(2*D,e);
|
|
|
|
// formatted output
|
|
std::cout << "% --------------------------- " << std::endl;
|
|
std::cout << "% --------------------------- " << std::endl << std::endl;
|
|
std::cout << "F=" << F << ";" << std::endl << std::endl;
|
|
|
|
} catch (const std::exception& e) {
|
|
std::cerr << "Cought exception " << e.what();
|
|
std::cerr << "in the main function of simple expression." << std::endl;
|
|
}
|
|
|
|
}
|