aft static method

Tensor aft(
  1. Tensor q,
  2. Tensor k,
  3. Tensor v,
  4. Tensor wb,
  5. bool masked,
)

Implementation

static Tensor aft(Tensor q, Tensor k, Tensor v, Tensor wb, bool masked) {
  return Tensor._raw(
    engine.aftForward(q._handle, k._handle, v._handle, wb._handle, masked),
    q.shape,
  );
}