Trait dfdx::nn::BuildModule

pub trait BuildModule<D: Device<E>, E: Dtype>: Sized + TensorCollection<E, D, To<E, D> = Self> {
    // Provided methods
    fn build(device: &D) -> Self { ... }
    fn try_build(device: &D) -> Result<Self, D::Err> { ... }
}

Something that can be built. Related to super::BuildOnDevice

Provided Methods

fn build(device: &D) -> Self

Construct it on the device

fn try_build(device: &D) -> Result<Self, D::Err>

Fallible version of BuildModule::build

Implementors

impl<D: Device<E>, E: Dtype, M: Sized + TensorCollection<E, D, To<E, D> = Self>> BuildModule<D, E> for M