Struct MlxClosureCustomVmap 

pub struct MlxClosureCustomVmap(/* private fields */);

Owned MLX custom VMAP closure, released exactly once.

Implementations

impl MlxClosureCustomVmap

pub fn from_fn<F>(callback: F) -> Result<Self, MlxCError>

where F: FnMut(&[MlxArray], &[i32]) -> Result<(Vec<MlxArray>, Vec<i32>), MlxCError> + 'static,

Builds an owned custom-VMAP closure from a Rust callable that receives inputs and input axes and returns outputs with output axes.

pub fn empty() -> Self

Creates an empty custom-VMAP closure handle.

pub fn set(&mut self, source: &Self) -> Result<(), MlxCError>

Copies the source closure handle into this handle.

pub fn apply( &self, inputs: &MlxArrayVector, input_axes: &[i32], ) -> Result<(MlxArrayVector, MlxVectorInt), MlxCError>

Applies the custom VMAP transform.

Errors

Returns the captured MLX-C description when application fails.

Trait Implementations

impl Debug for MlxClosureCustomVmap

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for MlxClosureCustomVmap

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.