mlx_rs::module

Trait UnaryModule

Source
pub trait UnaryModule: for<'a> Module<&'a Array, Output = Array> { }
Expand description

Marker trait for a unary neural network module.

This trait should not be implemented directly. Instead, implement Module with Args as a reference to the input.

Implementors§

Source§

impl<T> UnaryModule for T
where T: for<'a> Module<&'a Array, Output = Array>,