Function marian::bdot

• Defined in File expression_operators.cpp

Function Documentation

Expr marian::bdot(Expr a, Expr b, bool transA = false, bool transB = false, float scalar = 1.f)

Computes the batch dot product of a and b.

Computes \( C = \alpha \operatorname{op}(A) \cdot \operatorname{op}(B) \), where \( \operatorname{op}(A) = A \) if transA is false, and \( \operatorname{op}(A) = A^\top \) if true. The \(\alpha\) parameter is set by scalar.