Skip to main content

ClikaRT::ops::bmm

function

bmm()​

Tensor bmm(
    Tensor a,
    Tensor b,
    OptionalTensor bias = {},
    std::optional<Activation> activation = std::nullopt
)

Batched matrix multiply of two rank-3 tensors, with an optional fused bias and activation epilogue.

a [B, M, K] x b [B, K, N] -> [B, M, N], one independent matmul per batch index.

Parameters

  • a: left operand [B, M, K].
  • b: right operand [B, K, N].
  • bias: optional additive bias, broadcastable to the output.
  • activation: optional fused epilogue; absent = none.

Returns: [B, M, N].

Throws

  • ClikaRT::Error: (INVALID_ARGUMENT) when the operands are not rank-3 or the batch/inner dims disagree.

Declared in ClikaRT/compute/ops.h, line 2781