mlx-bun library API
Documentation
Preparing search index...
@mlx-bun/inference
@mlx-bun/inference/layers
createCausalMask
Function createCausalMask
createCausalMask
(
N
:
number
,
offset
:
number
,
windowSize
:
number
|
null
)
:
MlxArray
Port of
base.py
create_causal_mask (bool, [N, offset+N]).
Parameters
N
:
number
offset
:
number
windowSize
:
number
|
null
Returns
MlxArray
Settings
Member Visibility
Protected
Inherited
External
Theme
OS
Light
Dark
Documentation
mlx-bun library API
Loading...
Port of base.py create_causal_mask (bool, [N, offset+N]).