![]() |
Ginkgo Generated from branch based on main. Ginkgo version 1.10.0
A numerical linear algebra library targeting many-core architectures
|
Dense is a batch matrix format which explicitly stores all values of the matrix in each of the batches. More...
#include <ginkgo/core/matrix/batch_dense.hpp>
Public Types | |
| using | value_type = ValueType |
| using | index_type = int32 |
| using | transposed_type = Dense<ValueType> |
| using | unbatch_type = gko::matrix::Dense<ValueType> |
| using | absolute_type = remove_complex<Dense> |
| using | complex_type = to_complex<Dense> |
| Public Types inherited from gko::EnablePolymorphicAssignment< Dense< default_precision > > | |
| using | result_type |
| Public Types inherited from gko::ConvertibleTo< Dense< default_precision > > | |
| using | result_type |
| Public Types inherited from gko::ConvertibleTo< Dense< next_precision< default_precision > > > | |
| using | result_type |
Public Member Functions | |
| void | convert_to (Dense< next_precision< ValueType > > *result) const override |
| void | move_to (Dense< next_precision< ValueType > > *result) override |
| std::unique_ptr< unbatch_type > | create_view_for_item (size_type item_id) |
| Creates a mutable view (of gko::matrix::Dense type) of one item of the batch::matrix::Dense<value_type> object. | |
| std::unique_ptr< const unbatch_type > | create_const_view_for_item (size_type item_id) const |
| Creates a mutable view (of gko::matrix::Dense type) of one item of the batch::matrix::Dense<value_type> object. | |
| size_type | get_cumulative_offset (size_type batch_id) const |
| Get the cumulative storage size offset. | |
| value_type * | get_values () noexcept |
| Returns a pointer to the array of values of the multi-vector. | |
| const value_type * | get_const_values () const noexcept |
| Returns a pointer to the array of values of the multi-vector. | |
| value_type & | at (size_type batch_id, size_type row, size_type col) |
| Returns a single element for a particular batch item. | |
| value_type | at (size_type batch_id, size_type row, size_type col) const |
| Returns a single element for a particular batch item. | |
| ValueType & | at (size_type batch_id, size_type idx) noexcept |
| Returns a single element for a particular batch item. | |
| ValueType | at (size_type batch_id, size_type idx) const noexcept |
| Returns a single element for a particular batch item. | |
| value_type * | get_values_for_item (size_type batch_id) noexcept |
| Returns a pointer to the array of values of the matrix for a specific batch item. | |
| const value_type * | get_const_values_for_item (size_type batch_id) const noexcept |
| Returns a pointer to the array of values of the matrix for a specific batch item. | |
| size_type | get_num_stored_elements () const noexcept |
| Returns the number of elements explicitly stored in the batch matrix, cumulative across all the batch items. | |
| size_type | get_num_elements_per_item () const noexcept |
| Returns the number of stored elements in each batch item. | |
| Dense * | apply (ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x) |
| Apply the matrix to a multi-vector. | |
| Dense * | apply (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const MultiVector< value_type > > b, ptr_param< const MultiVector< value_type > > beta, ptr_param< MultiVector< value_type > > x) |
| Apply the matrix to a multi-vector with a linear combination of the given input vector. | |
| const Dense * | apply (ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x) const |
| const Dense * | apply (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const MultiVector< value_type > > b, ptr_param< const MultiVector< value_type > > beta, ptr_param< MultiVector< value_type > > x) const |
| void | scale (const array< value_type > &row_scale, const array< value_type > &col_scale) |
| Performs in-place row and column scaling for this matrix. | |
| void | scale_add (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const batch::matrix::Dense< value_type > > b) |
| Performs the operation this = alpha*this + b. | |
| void | add_scaled_identity (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const MultiVector< value_type > > beta) |
| Performs the operation this = alpha*I + beta*this. | |
| Public Member Functions inherited from gko::EnableAbstractPolymorphicObject< Dense< default_precision >, BatchLinOp > | |
| std::unique_ptr< Dense< default_precision > > | create_default (std::shared_ptr< const Executor > exec) const |
| std::unique_ptr< Dense< default_precision > > | clone (std::shared_ptr< const Executor > exec) const |
| Dense< default_precision > * | copy_from (const PolymorphicObject *other) |
| Dense< default_precision > * | move_from (ptr_param< PolymorphicObject > other) |
| Dense< default_precision > * | clear () |
| Public Member Functions inherited from gko::EnablePolymorphicAssignment< Dense< default_precision > > | |
| void | convert_to (result_type *result) const override |
| void | move_to (result_type *result) override |
| Public Member Functions inherited from gko::ConvertibleTo< Dense< next_precision< default_precision > > > | |
| virtual void | convert_to (result_type *result) const=0 |
| Converts the implementer to an object of type result_type. | |
| virtual void | move_to (result_type *result)=0 |
| Converts the implementer to an object of type result_type by moving data from this object. | |
Static Public Member Functions | |
| static std::unique_ptr< Dense > | create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size=batch_dim< 2 >{}) |
| Creates an uninitialized Dense matrix of the specified size. | |
| static std::unique_ptr< Dense > | create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size, array< value_type > values) |
| Creates a Dense matrix from an already allocated (and initialized) array. | |
| template<typename InputValueType> | |
| static std::unique_ptr< Dense > | create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size, std::initializer_list< InputValueType > values) |
| create(std::shared_ptr<const Executor>,const batch_dim<2>&, array<value_type>) | |
| static std::unique_ptr< const Dense > | create_const (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &sizes, gko::detail::const_array_view< ValueType > &&values) |
| Creates a constant (immutable) batch dense matrix from a constant array. | |
Friends | |
| class | EnablePolymorphicObject< Dense, BatchLinOp > |
| class | Dense< to_complex< ValueType > > |
| class | Dense< previous_precision< ValueType > > |
Dense is a batch matrix format which explicitly stores all values of the matrix in each of the batches.
The values in each of the batches are stored in row-major format (values belonging to the same row appear consecutive in the memory and the values of each batch item are also stored consecutively in memory).
| ValueType | precision of matrix elements |
| void gko::batch::matrix::Dense< ValueType >::add_scaled_identity | ( | ptr_param< const MultiVector< value_type > > | alpha, |
| ptr_param< const MultiVector< value_type > > | beta ) |
Performs the operation this = alpha*I + beta*this.
| alpha | the scalar for identity |
| beta | the scalar to multiply this matrix |
| Dense * gko::batch::matrix::Dense< ValueType >::apply | ( | ptr_param< const MultiVector< value_type > > | alpha, |
| ptr_param< const MultiVector< value_type > > | b, | ||
| ptr_param< const MultiVector< value_type > > | beta, | ||
| ptr_param< MultiVector< value_type > > | x ) |
Apply the matrix to a multi-vector with a linear combination of the given input vector.
Represents the matrix vector multiplication, x = alpha * A
| alpha | the scalar to scale the matrix-vector product with |
| b | the multi-vector to be applied to |
| beta | the scalar to scale the x vector with |
| x | the output multi-vector |
| const Dense * gko::batch::matrix::Dense< ValueType >::apply | ( | ptr_param< const MultiVector< value_type > > | alpha, |
| ptr_param< const MultiVector< value_type > > | b, | ||
| ptr_param< const MultiVector< value_type > > | beta, | ||
| ptr_param< MultiVector< value_type > > | x ) const |
| Dense * gko::batch::matrix::Dense< ValueType >::apply | ( | ptr_param< const MultiVector< value_type > > | b, |
| ptr_param< MultiVector< value_type > > | x ) |
Apply the matrix to a multi-vector.
Represents the matrix vector multiplication, x = A * b, where x and b are both multi-vectors.
| b | the multi-vector to be applied to |
| x | the output multi-vector |
| const Dense * gko::batch::matrix::Dense< ValueType >::apply | ( | ptr_param< const MultiVector< value_type > > | b, |
| ptr_param< MultiVector< value_type > > | x ) const |
|
inlinenoexcept |
Returns a single element for a particular batch item.
| batch_id | the batch item index to be queried |
| row | the row of the requested element |
| col | the column of the requested element |
|
inlinenoexcept |
Returns a single element for a particular batch item.
Useful for iterating across all elements of the matrix. However, it is less efficient than the two-parameter variant of this method.
| batch_id | the batch item index to be queried |
| idx | a linear index of the requested element |
|
inline |
Returns a single element for a particular batch item.
| batch_id | the batch item index to be queried |
| row | the row of the requested element |
| col | the column of the requested element |
|
inline |
Returns a single element for a particular batch item.
| batch_id | the batch item index to be queried |
| row | the row of the requested element |
| col | the column of the requested element |
|
static |
Creates a Dense matrix from an already allocated (and initialized) array.
| exec | Executor associated to the matrix |
| size | sizes of the batch matrices in a batch_dim object |
| values | array of matrix values |
|
inlinestatic |
|
static |
|
static |
Creates a constant (immutable) batch dense matrix from a constant array.
| exec | the executor to create the matrix on |
| size | the dimensions of the matrix |
| values | the value array of the matrix |
| std::unique_ptr< const unbatch_type > gko::batch::matrix::Dense< ValueType >::create_const_view_for_item | ( | size_type | item_id | ) | const |
Creates a mutable view (of gko::matrix::Dense type) of one item of the batch::matrix::Dense<value_type> object.
Does not perform any deep copies, but only returns a view of the data.
| item_id | The index of the batch item |
| std::unique_ptr< unbatch_type > gko::batch::matrix::Dense< ValueType >::create_view_for_item | ( | size_type | item_id | ) |
Creates a mutable view (of gko::matrix::Dense type) of one item of the batch::matrix::Dense<value_type> object.
Does not perform any deep copies, but only returns a view of the data.
| item_id | The index of the batch item |
|
inlinenoexcept |
Returns a pointer to the array of values of the multi-vector.
|
inlinenoexcept |
Returns a pointer to the array of values of the matrix for a specific batch item.
| batch_id | the id of the batch item. |
|
inline |
Get the cumulative storage size offset.
| batch_id | the batch id |
Referenced by get_const_values_for_item(), and get_values_for_item().
|
inlinenoexcept |
Returns the number of stored elements in each batch item.
|
inlinenoexcept |
Returns the number of elements explicitly stored in the batch matrix, cumulative across all the batch items.
Referenced by get_num_elements_per_item().
|
inlinenoexcept |
Returns a pointer to the array of values of the multi-vector.
|
inlinenoexcept |
Returns a pointer to the array of values of the matrix for a specific batch item.
| batch_id | the id of the batch item. |
| void gko::batch::matrix::Dense< ValueType >::scale | ( | const array< value_type > & | row_scale, |
| const array< value_type > & | col_scale ) |
Performs in-place row and column scaling for this matrix.
| row_scale | the row scalars |
| col_scale | the column scalars |
| void gko::batch::matrix::Dense< ValueType >::scale_add | ( | ptr_param< const MultiVector< value_type > > | alpha, |
| ptr_param< const batch::matrix::Dense< value_type > > | b ) |
Performs the operation this = alpha*this + b.
| alpha | the scalar to multiply this matrix |
| b | the matrix to add |