call method

MlxArray call(
  1. MlxArray tokenIds
)

Implementation

MlxArray call(MlxArray tokenIds) => mx.take(weight, tokenIds, axis: 0);