Welcome to our Support Center

gemm

Description

This function performs the matrix-matrix multiplication : C = Ξ± * op(A) * op(B) + Ξ² * C
Type : polymorphic.

 

Input parameters

 

Β A : class, 2D tensor of dimension M x K.
Β B : class, 2D tensor of dimension K x N.
Β C : class, 2D tensor of dimension M x N.
Op(A) : enum, operation that is non- or transpose.

    • CUBLAS_OP_N : the non-transpose operation is selected
    • CUBLAS_OP_T : the transpose operation is selected
    • CUBLAS_OP_H : the conjugate transpose operation is selected

Op(B) : enum, operation that is non- or transpose.

    • CUBLAS_OP_N : the non-transpose operation is selected
    • CUBLAS_OP_T : the transpose operation is selected
    • CUBLAS_OP_H : the conjugate transpose operation is selected

Ξ±Β : float, scalar used for multiplication.
Ξ²Β : float, scalar used for multiplication, if beta==0 then C does not have to be a valid input.

Output parameters

 

Β C : class, 2D tensor of dimension M x N.

Example

All these exemples are snippets PNG, you can drop these Snippet onto the block diagram and get the depicted code added to your VI (Do not forget to install PERRINE library to run it).

The non-transposed operation is selected for A and B

The transposed operation is selected for A and B

The transposed operation is selected for B

The transposed operation is selected for A

Table of Contents
Index