tensorium::Tensor< K, Rank > Class Template Reference

Multi-dimensional tensor class with fixed rank and SIMD support. More...

#include <Tensor.hpp>

Collaboration diagram for tensorium::Tensor< K, Rank >:

Public Types
using	value_type = K

Public Member Functions
	Tensor ()
	Default constructor.
	Tensor (const std::array< size_t, Rank > &dims)
	Construct tensor with given dimensions.
size_t	flatten_index (size_t i, size_t j, size_t k, size_t l) const
std::array< size_t, Rank >	shape () const
void	update_strides ()
void	fill (K value)
	Fill tensor with a constant value.
void	resize (const std::array< size_t, 2 > &dims)
	Resize 2D tensor.
void	resize (size_t d0, size_t d1)
	Resize 2D tensor.
Element Access
void	resize (size_t d0, size_t d1, size_t d2)
K &	operator() (const std::array< size_t, Rank > &indices)
const K &	operator() (const std::array< size_t, Rank > &indices) const
const K &	operator() (size_t i, size_t j, size_t k, size_t l) const
K &	operator() (size_t i, size_t j)
K &	operator() (size_t i, size_t j, size_t k, size_t l)
const K &	operator() (size_t i, size_t j) const
K &	operator() (size_t i, size_t j, size_t k)
const K &	operator() (size_t i, size_t j, size_t k) const
Debug and Utilities
void	print_shape () const
	Print the shape (dimensions) of the tensor.
void	print () const
	Print a 2D tensor (Rank == 2)

Public Attributes
std::array< size_t, Rank >	dimensions
	Dimensions of the tensor (e.g., {4,4,4,4})
size_t	total_size
aligned_vector< K >	data
size_t	block_size
std::array< size_t, Rank >	strides

basic tensor operations (Rank == 2)
using	reg = typename Simd::reg
const size_t *strides	const
constexpr size_t	W = Simd::width / sizeof(size_t)
size_t	acc = 0
size_t	i = 0
return	acc
	__attribute__ ((always_inline, hot, flatten)) inline size_t flatten_index_simd(const size_t *indices
	Convert a multi-index into a flattened linear index using SIMD.
	for (;i+W - 1< Rank;i+=W)
	for (;i< Rank;++i) acc+
__attribute__((always_inline, hot, flatten)) inline size_t flatten_index(const std	__attribute__ ((always_inline, hot, flatten)) Tensor< K
	Convert multi-index to flat index.
__attribute__((always_inline, hot, flatten)) inline size_t flatten_index(const std	transpose_simd () const
template<size_t R1, size_t R2>
static Tensor< K, R1+R2 >	tensor_product (const Tensor< K, R1 > &A, const Tensor< K, R2 > &B)
	Compute the tensor (outer) product of two tensors.

Detailed Description

template<typename K, std::size_t Rank>
class tensorium::Tensor< K, Rank >

Multi-dimensional tensor class with fixed rank and SIMD support.

This class provides high-performance operations on tensors of arbitrary rank, supporting element-wise access, contraction, transposition, and tensor product, all implemented with SIMD optimization and aligned memory.

Template Parameters

K	Scalar type (e.g., float or double)
Rank	Number of tensor dimensions

Member Typedef Documentation

reg

template<typename K , std::size_t Rank>

using tensorium::Tensor< K, Rank >::reg = typename Simd::reg

value_type

template<typename K , std::size_t Rank>

using tensorium::Tensor< K, Rank >::value_type = K

Constructor & Destructor Documentation

Tensor() [1/2]

template<typename K , std::size_t Rank>

tensorium::Tensor< K, Rank >::Tensor ( )

inline

Default constructor.

Referenced by tensorium::Tensor< K, Rank >::tensor_product(), and tensorium::Tensor< K, Rank >::transpose_simd().

Here is the caller graph for this function:

Tensor() [2/2]

template<typename K , std::size_t Rank>

tensorium::Tensor< K, Rank >::Tensor ( const std::array< size_t, Rank > & dims )

inline

Construct tensor with given dimensions.

Parameters

dims	Dimension sizes

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::dimensions, tensorium::Tensor< K, Rank >::i, tensorium::Tensor< K, Rank >::strides, and tensorium::Tensor< K, Rank >::total_size.

Member Function Documentation

__attribute__() [1/2]

template<typename K , std::size_t Rank>

tensorium::Tensor< K, Rank >::__attribute__

(

(always_inline, hot, flatten)

)

const

Convert a multi-index into a flattened linear index using SIMD.

Parameters

indices	Array of indices
strides	Array of strides

Returns

Flattened index

__attribute__() [2/2]

template<typename K , std::size_t Rank>

__attribute__((always_inline, hot, flatten)) inline size_t flatten_index(const std tensorium::Tensor< K, Rank >::__attribute__

(

(always_inline, hot, flatten)

)

Convert multi-index to flat index.

Parameters

indices

Array of indices

Returns

Flattened index

Perform contraction over two indices using SIMD

Contracts a tensor \( T_{i_1 \dots i_k i j i_{k+1} \dots i_n} \) to \( T'_{i_1 \dots i_k i_{k+1} \dots i_n} \)

Parameters

t	Input tensor
i	First index to contract
j	Second index to contract

Returns

Contracted tensor of rank (Rank - 2)

Transpose a 2D tensor using SIMD

Only valid for tensors of rank 2.

Returns

Transposed tensor

fill()

template<typename K , std::size_t Rank>

void tensorium::Tensor< K, Rank >::fill ( K value )

inline

Fill tensor with a constant value.

References tensorium::Tensor< K, Rank >::data.

Referenced by tensorium_RG::RiemannTensor< T >::compute(), tensorium::SpectalChebyshev< T >::compute(), tensorium_RG::Metric< T >::compute_flrw(), tensorium_RG::Metric< T >::compute_kerr(), tensorium_RG::Metric< T >::compute_kerr_schild(), tensorium_RG::Metric< T >::compute_minkowski(), tensorium_RG::Metric< T >::compute_schwarzschild(), and tensorium_RG::RicciTensor< T >::contract_to_ricci().

Here is the caller graph for this function:

flatten_index()

template<typename K , std::size_t Rank>

size_t tensorium::Tensor< K, Rank >::flatten_index ( size_t i, size_t j, size_t k, size_t l ) const

inline

References tensorium::Tensor< K, Rank >::flatten_index(), and tensorium::Tensor< K, Rank >::i.

Referenced by tensorium::Tensor< K, Rank >::flatten_index(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), and tensorium::Tensor< K, Rank >::operator()().

Here is the call graph for this function:

Here is the caller graph for this function:

for() [1/2]

template<typename K , std::size_t Rank>

tensorium::Tensor< K, Rank >::for ( )

inline

References tensorium::Tensor< K, Rank >::acc, tensorium::Tensor< K, Rank >::i, and tensorium::Tensor< K, Rank >::strides.

for() [2/2]

template<typename K , std::size_t Rank>

tensorium::Tensor< K, Rank >::for

(

)

operator()() [1/8]

template<typename K , std::size_t Rank>

K & tensorium::Tensor< K, Rank >::operator() ( const std::array< size_t, Rank > & indices )

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::flatten_index(), and tensorium::Tensor< K, Rank >::total_size.

Here is the call graph for this function:

operator()() [2/8]

template<typename K , std::size_t Rank>

const K & tensorium::Tensor< K, Rank >::operator() ( const std::array< size_t, Rank > & indices ) const

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::flatten_index(), and tensorium::Tensor< K, Rank >::total_size.

Here is the call graph for this function:

operator()() [3/8]

template<typename K , std::size_t Rank>

K & tensorium::Tensor< K, Rank >::operator() ( size_t i, size_t j )

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::dimensions, and tensorium::Tensor< K, Rank >::i.

operator()() [4/8]

template<typename K , std::size_t Rank>

const K & tensorium::Tensor< K, Rank >::operator() ( size_t i, size_t j ) const

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::dimensions, and tensorium::Tensor< K, Rank >::i.

operator()() [5/8]

template<typename K , std::size_t Rank>

K & tensorium::Tensor< K, Rank >::operator() ( size_t i, size_t j, size_t k )

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::dimensions, and tensorium::Tensor< K, Rank >::i.

operator()() [6/8]

template<typename K , std::size_t Rank>

const K & tensorium::Tensor< K, Rank >::operator() ( size_t i, size_t j, size_t k ) const

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::dimensions, and tensorium::Tensor< K, Rank >::i.

operator()() [7/8]

template<typename K , std::size_t Rank>

K & tensorium::Tensor< K, Rank >::operator() ( size_t i, size_t j, size_t k, size_t l )

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::flatten_index(), and tensorium::Tensor< K, Rank >::i.

Here is the call graph for this function:

operator()() [8/8]

template<typename K , std::size_t Rank>

const K & tensorium::Tensor< K, Rank >::operator() ( size_t i, size_t j, size_t k, size_t l ) const

inline

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::flatten_index(), and tensorium::Tensor< K, Rank >::i.

Here is the call graph for this function:

print()

template<typename K , std::size_t Rank>

void tensorium::Tensor< K, Rank >::print ( ) const

inline

Print a 2D tensor (Rank == 2)

References tensorium::Tensor< K, Rank >::dimensions, and tensorium::Tensor< K, Rank >::i.

print_shape()

template<typename K , std::size_t Rank>

void tensorium::Tensor< K, Rank >::print_shape ( ) const

inline

Print the shape (dimensions) of the tensor.

References tensorium::Tensor< K, Rank >::dimensions, and tensorium::Tensor< K, Rank >::i.

resize() [1/3]

template<typename K , std::size_t Rank>

void tensorium::Tensor< K, Rank >::resize ( const std::array< size_t, 2 > & dims )

inline

Resize 2D tensor.

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::dimensions, tensorium::Tensor< K, Rank >::total_size, and tensorium::Tensor< K, Rank >::update_strides().

Referenced by tensorium_RG::BSSNChristoffel< T >::compute(), tensorium_RG::TildeGamma< T >::compute(), tensorium::SpectalChebyshev< T >::compute(), tensorium_RG::Metric< T >::compute_conformal_metric(), tensorium_RG::Metric< T >::compute_flrw(), tensorium_RG::Metric< T >::compute_kerr(), tensorium_RG::Metric< T >::compute_kerr_schild(), tensorium_RG::Metric< T >::compute_minkowski(), tensorium_RG::compute_partial_derivatives_3D(), tensorium_RG::compute_partial_derivatives_tensor2D(), tensorium_RG::compute_partial_derivatives_vector(), tensorium_RG::Metric< T >::compute_schwarzschild(), tensorium::Tensor< K, Rank >::resize(), and tensorium::Vector< K >::resize().

Here is the call graph for this function:

Here is the caller graph for this function:

resize() [2/3]

template<typename K , std::size_t Rank>

void tensorium::Tensor< K, Rank >::resize ( size_t d0, size_t d1 )

inline

Resize 2D tensor.

References tensorium::Tensor< K, Rank >::resize().

Here is the call graph for this function:

resize() [3/3]

template<typename K , std::size_t Rank>

void tensorium::Tensor< K, Rank >::resize ( size_t d0, size_t d1, size_t d2 )

inline

References tensorium::Tensor< K, Rank >::data, and tensorium::Tensor< K, Rank >::dimensions.

shape()

template<typename K , std::size_t Rank>

std::array< size_t, Rank > tensorium::Tensor< K, Rank >::shape ( ) const

inline

References tensorium::Tensor< K, Rank >::dimensions.

Referenced by tensorium_RG::BSSNChristoffel< T >::compute(), tensorium_RG::TildeGamma< T >::compute(), tensorium_RG::compute_partial_derivatives_3D(), and tensorium::Tensor< K, Rank >::tensor_product().

Here is the caller graph for this function:

tensor_product()

template<typename K , std::size_t Rank>

template<size_t R1, size_t R2>

static Tensor< K, R1+R2 > tensorium::Tensor< K, Rank >::tensor_product ( const Tensor< K, R1 > & A, const Tensor< K, R2 > & B )

inlinestatic

Compute the tensor (outer) product of two tensors.

Resulting tensor has rank = R1 + R2.

\[ (A \otimes B)_{i_1 \dots i_R} = A_{i_1 \dots i_{R_1}} \cdot B_{i_{R_1 + 1} \dots i_R} \]

Template Parameters

R1	Rank of tensor A
R2	Rank of tensor B

Parameters

A	Tensor A
B	Tensor B

Returns

Tensor product \( A \otimes B \)

References tensorium::Tensor< K, Rank >::data, tensorium::Tensor< K, Rank >::dimensions, tensorium::Tensor< K, Rank >::i, tensorium::Tensor< K, Rank >::shape(), tensorium::Tensor< K, Rank >::Tensor(), tensorium::Tensor< K, Rank >::total_size, and tensorium::Tensor< K, Rank >::W.

Here is the call graph for this function:

transpose_simd()

template<typename K , std::size_t Rank>

__attribute__((always_inline, hot, flatten)) inline size_t flatten_index(const std tensorium::Tensor< K, Rank >::transpose_simd ( ) const

inline

References tensorium::Tensor< K, Rank >::dimensions, tensorium::Tensor< K, Rank >::i, tensorium::Tensor< K, Rank >::Tensor(), and tensorium::Tensor< K, Rank >::W.

Referenced by tensorium::transpose_tensor().

Here is the call graph for this function:

Here is the caller graph for this function:

update_strides()

template<typename K , std::size_t Rank>

void tensorium::Tensor< K, Rank >::update_strides ( )

inline

References tensorium::Tensor< K, Rank >::dimensions, tensorium::Tensor< K, Rank >::i, and tensorium::Tensor< K, Rank >::strides.

Referenced by tensorium::Tensor< K, Rank >::resize().

Here is the caller graph for this function:

Member Data Documentation

acc [1/2]

template<typename K , std::size_t Rank>

size_t tensorium::Tensor< K, Rank >::acc = 0

Referenced by tensorium::Tensor< K, Rank >::for().

acc [2/2]

template<typename K , std::size_t Rank>

return tensorium::Tensor< K, Rank >::acc

block_size

template<typename K , std::size_t Rank>

size_t tensorium::Tensor< K, Rank >::block_size

const

template<typename K , std::size_t Rank>

const size_t* strides tensorium::Tensor< K, Rank >::const

Initial value:

{
                        using Simd = simd::SimdTraits<size_t, DefaultISA>

data

template<typename K , std::size_t Rank>

aligned_vector<K> tensorium::Tensor< K, Rank >::data

Referenced by tensorium::Vector< K >::__attribute__(), tensorium::Vector< K >::__attribute__(), tensorium::Vector< K >::__attribute__(), tensorium::Vector< K >::__attribute__(), tensorium::Vector< K >::__attribute__(), tensorium::Tensor< K, Rank >::fill(), tensorium::Vector< K >::for(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::resize(), tensorium::Tensor< K, Rank >::resize(), tensorium::Tensor< K, Rank >::Tensor(), and tensorium::Tensor< K, Rank >::tensor_product().

dimensions

template<typename K , std::size_t Rank>

std::array<size_t, Rank> tensorium::Tensor< K, Rank >::dimensions

Dimensions of the tensor (e.g., {4,4,4,4})

Referenced by tensorium_RG::__attribute__(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::print(), tensorium::Tensor< K, Rank >::print_shape(), tensorium::Tensor< K, Rank >::resize(), tensorium::Tensor< K, Rank >::resize(), tensorium::Tensor< K, Rank >::shape(), tensorium::Tensor< K, Rank >::Tensor(), tensorium::Tensor< K, Rank >::tensor_product(), tensorium::tensor_to_matrix(), tensorium::Tensor< K, Rank >::transpose_simd(), and tensorium::Tensor< K, Rank >::update_strides().

i

template<typename K , std::size_t Rank>

size_t tensorium::Tensor< K, Rank >::i = 0

Referenced by tensorium::Tensor< K, Rank >::flatten_index(), tensorium::Tensor< K, Rank >::for(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::print(), tensorium::Tensor< K, Rank >::print_shape(), tensorium::Tensor< K, Rank >::Tensor(), tensorium::Tensor< K, Rank >::tensor_product(), tensorium::Tensor< K, Rank >::transpose_simd(), and tensorium::Tensor< K, Rank >::update_strides().

strides

template<typename K , std::size_t Rank>

std::array<size_t, Rank> tensorium::Tensor< K, Rank >::strides

Referenced by tensorium::Tensor< K, Rank >::for(), tensorium::Tensor< K, Rank >::Tensor(), and tensorium::Tensor< K, Rank >::update_strides().

total_size

template<typename K , std::size_t Rank>

size_t tensorium::Tensor< K, Rank >::total_size

Referenced by tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::operator()(), tensorium::Tensor< K, Rank >::resize(), tensorium::Tensor< K, Rank >::Tensor(), and tensorium::Tensor< K, Rank >::tensor_product().

W

template<typename K , std::size_t Rank>

constexpr size_t tensorium::Tensor< K, Rank >::W = Simd::width / sizeof(size_t)

constexpr

Referenced by tensorium::Tensor< K, Rank >::tensor_product(), and tensorium::Tensor< K, Rank >::transpose_simd().

---

The documentation for this class was generated from the following file:

• includes/Tensorium/Core/Tensor.hpp