5#ifndef GKO_PUBLIC_CORE_MATRIX_BATCH_IDENTITY_HPP_
6#define GKO_PUBLIC_CORE_MATRIX_BATCH_IDENTITY_HPP_
9#include <ginkgo/core/base/batch_lin_op.hpp>
10#include <ginkgo/core/base/batch_multi_vector.hpp>
11#include <ginkgo/core/base/executor.hpp>
12#include <ginkgo/core/base/types.hpp>
13#include <ginkgo/core/base/utils.hpp>
14#include <ginkgo/core/matrix/identity.hpp>
31template <
typename ValueType = default_precision>
36 using value_type = ValueType;
37 using index_type =
int32;
90 static std::unique_ptr<Identity>
create(
91 std::shared_ptr<const Executor> exec,
95 Identity(std::shared_ptr<const Executor> exec,
This mixin inherits from (a subclass of) PolymorphicObject and provides a base implementation of a ne...
Definition polymorphic_object.hpp:662
Definition batch_lin_op.hpp:59
The EnableBatchLinOp mixin can be used to provide sensible default implementations of the majority of...
Definition batch_lin_op.hpp:252
MultiVector stores multiple vectors in a batched fashion and is useful for batched operations.
Definition batch_multi_vector.hpp:59
The batch Identity matrix, which represents a batch of Identity matrices.
Definition batch_identity.hpp:32
Identity * 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 Identity * 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
Identity * apply(ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x)
Apply the matrix to a multi-vector.
static std::unique_ptr< Identity > create(std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size=batch_dim< 2 >{})
Creates an Identity matrix of the specified size.
const Identity * apply(ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x) const
This class is a utility which efficiently implements the identity matrix (a linear operator which map...
Definition identity.hpp:35
This class is used for function parameters in the place of raw pointers.
Definition utils_helper.hpp:41
The Ginkgo namespace.
Definition abstract_factory.hpp:20
typename detail::remove_complex_s< T >::type remove_complex
Obtain the type which removed the complex of complex/scalar type or the template parameter of class b...
Definition math.hpp:260
std::int32_t int32
32-bit signed integral type.
Definition types.hpp:106
typename detail::to_complex_s< T >::type to_complex
Obtain the type which adds the complex of complex/scalar type or the template parameter of class by a...
Definition math.hpp:279
A type representing the dimensions of a multidimensional batch object.
Definition batch_dim.hpp:27