Skip to main content

ClikaRT::ops::bmm

function

bmm()​

Tensor bmm(
    Tensor input,
    Tensor other,
    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

  • input: left operand [B, M, K].
  • other: 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 2801