Matrix Operations#

Transpose#

#include <raft/linalg/transpose.cuh>

namespace raft::linalg

template<typename T, typename IndexType, typename LayoutPolicy, typename AccessorPolicy>
auto transpose(raft::resources const &handle, raft::mdspan<T, raft::matrix_extent<IndexType>, LayoutPolicy, AccessorPolicy> in, raft::mdspan<T, raft::matrix_extent<IndexType>, LayoutPolicy, AccessorPolicy> out) -> std::enable_if_t<std::is_floating_point_v<T>, void>#

Transpose a matrix. The output has same layout policy as the input.

Template Parameters:

  • T – Data type of input matrix element.

  • IndexType – Index type of matrix extent.

  • LayoutPolicy – Layout type of the input matrix. When layout is strided, it can be a submatrix of a larger matrix. Arbitrary stride is not supported.

  • AccessorPolicy – Accessor for the input and output, must be valid accessor on device.

Parameters:

  • handle – [in] raft handle for managing expensive cuda resources.

  • in – [in] Input matrix.

  • out – [out] Output matrix, storage is pre-allocated by caller.