aftCross static method

Tensor aftCross(
  1. Tensor q,
  2. Tensor k,
  3. Tensor v,
  4. Tensor wb,
)

Implementation

static Tensor aftCross(Tensor q, Tensor k, Tensor v, Tensor wb) {
  return Tensor._raw(
    engine.aftCrossForward(q._handle, k._handle, v._handle, wb._handle),
    q.shape,
  );
}