oa::FnMatrix::mergeHeads

mergeHeads: [B*H,S,D/H] -> [B*S,D]. exact inverse of splitHeads.

FnC++ public headers

Function Documentation

Matrix oa::FnMatrix::mergeHeads(
	const Matrix & inX,
	oa::I32 inBatch,
	oa::I32 inSeqLen,
	oa::I32 inNumHeads
)

mergeHeads: [B*H,S,D/H] -> [B*S,D]. exact inverse of splitHeads.

inX

const Matrix &

—

inBatch

oa::I32

—

inSeqLen

oa::I32

—

inNumHeads

oa::I32

—

Matrix

The declared return value.