Class MatMul

Defined in File mat_mul.h

Inheritance Relationships

Base Type

public mindspore::ops::PrimitiveC (Class PrimitiveC)

Derived Type

public mindspore::ops::MatMulFusion (Class MatMulFusion)

Class Documentation

class MatMul : public mindspore::ops::PrimitiveC

Multiplies matrix a and matrix b. Refer to Python API mindspore.ops.MatMul for more details.

Subclassed by mindspore::ops::MatMulFusion

Public Functions

inline MatMul(): Constructor.

~MatMul() = default: Destructor.

void Init(bool transpose_a = false, bool transpose_b = false): Init. Refer to the parameters of Python API mindspore.ops.MatMul for the inputs.

void set_transpose_a(bool transpose_a): Set transpose_a.

void set_transpose_b(bool transpose_b): Set transpose_b.

bool get_transpose_a() const

Get transpose_a.

Returns: transpose_a.

bool get_transpose_b() const

Get transpose_b.

Returns: transpose_b.