//clika-runtime/io.clika.runtime/Ops/bmm

bmm

[common]
fun bmm(input: Tensor, other: Tensor, bias: Tensor? = null, activation: Activation? = null): Tensor

bmm(input: Tensor, other: Tensor, bias: Tensor? = null, activation: Activation? = null): the bmm operator. 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.