Class BatchMatmul

• Defined in File batch_matmul.h

Inheritance Relationships

Base Type

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

Class Documentation

class BatchMatmul : public mindspore::ops::PrimitiveC

Computes matrix multiplication between two tensors by batch. Refer to Python API mindspore.ops.BatchMatmul for more details.

Public Functions

inline BatchMatmul()

Constructor.

~BatchMatmul() = default

Destructor.

void Init(bool transpose_a = false, bool transpose_b = false)

Init. Refer to the parameters of Python API mindspore.ops.BatchMatmul 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.