Trait border::agent::tch::model::base::Model2

pub trait Model2: ModelBase {
    type Input1;
    type Input2;
    type Output;
    fn forward(&self, x1s: &Self::Input1, x2s: &Self::Input2) -> Self::Output;
}

Neural networks with double inputs and a single output.

Associated Types

type Input1

An input of the neural network.

type Input2

The other input of the neural network.

type Output

The output of the neural network.

Loading content...

Required methods

fn forward(&self, x1s: &Self::Input1, x2s: &Self::Input2) -> Self::Output

Performs forward computation given a pair of inputs.

Loading content...

Implementors

impl Model2 for Model2_1

type Input1 = Tensor

type Input2 = Tensor

type Output = Tensor

fn forward(&self, x1s: &Tensor, x2s: &Tensor) -> Tensor

Loading content...