Trait lfa::transforms::Transform

pub trait Transform<T: ?Sized> {
type Output;
    fn transform(&self, x: T) -> Self::Output;
    fn grad(&self, x: T) -> T;
}

An interface for differentiable transformations.

Associated Types

type Output

Loading content...

Required methods

fn transform(&self, x: T) -> Self::Output

Return the value of the transform for input x.

fn grad(&self, x: T) -> T

Return the gradient of the transform for input x.

Loading content...

Implementors

impl Transform<[f64; 2]> for Tanh

type Output = [f64; 2]

fn transform(&self, x: [f64; 2]) -> [f64; 2]

fn grad(&self, x: [f64; 2]) -> [f64; 2]

impl Transform<[f64; 2]> for Identity

type Output = [f64; 2]

fn transform(&self, x: [f64; 2]) -> [f64; 2]

fn grad(&self, _: [f64; 2]) -> [f64; 2]

impl Transform<[f64; 2]> for Softplus

type Output = [f64; 2]

fn transform(&self, x: [f64; 2]) -> [f64; 2]

fn grad(&self, x: [f64; 2]) -> [f64; 2]

impl Transform<[f64; 2]> for LogSumExp

type Output = f64

fn transform(&self, x: [f64; 2]) -> f64

fn grad(&self, x: [f64; 2]) -> [f64; 2]

impl Transform<[f64; 2]> for Logistic

type Output = [f64; 2]

fn transform(&self, x: [f64; 2]) -> [f64; 2]

fn grad(&self, x: [f64; 2]) -> [f64; 2]

impl Transform<[f64; 2]> for Exp

type Output = [f64; 2]

fn transform(&self, x: [f64; 2]) -> [f64; 2]

fn grad(&self, x: [f64; 2]) -> [f64; 2]

impl Transform<[f64; 3]> for Tanh

type Output = [f64; 3]

fn transform(&self, x: [f64; 3]) -> [f64; 3]

fn grad(&self, x: [f64; 3]) -> [f64; 3]

impl Transform<[f64; 3]> for Identity

type Output = [f64; 3]

fn transform(&self, x: [f64; 3]) -> [f64; 3]

fn grad(&self, _: [f64; 3]) -> [f64; 3]

impl Transform<[f64; 3]> for Softplus

type Output = [f64; 3]

fn transform(&self, x: [f64; 3]) -> [f64; 3]

fn grad(&self, x: [f64; 3]) -> [f64; 3]

impl Transform<[f64; 3]> for LogSumExp

type Output = f64

fn transform(&self, x: [f64; 3]) -> f64

fn grad(&self, x: [f64; 3]) -> [f64; 3]

impl Transform<[f64; 3]> for Logistic

type Output = [f64; 3]

fn transform(&self, x: [f64; 3]) -> [f64; 3]

fn grad(&self, x: [f64; 3]) -> [f64; 3]

impl Transform<[f64; 3]> for Exp

type Output = [f64; 3]

fn transform(&self, x: [f64; 3]) -> [f64; 3]

fn grad(&self, x: [f64; 3]) -> [f64; 3]

impl Transform<f64> for Tanh

type Output = f64

fn transform(&self, x: f64) -> f64

fn grad(&self, x: f64) -> f64

impl Transform<f64> for Identity

type Output = f64

fn transform(&self, x: f64) -> f64

fn grad(&self, _: f64) -> f64

impl Transform<f64> for Softplus

type Output = f64

fn transform(&self, x: f64) -> f64

fn grad(&self, x: f64) -> f64

impl Transform<f64> for LogSumExp

type Output = f64

fn transform(&self, x: f64) -> f64

fn grad(&self, x: f64) -> f64

impl Transform<f64> for Logistic

type Output = f64

fn transform(&self, x: f64) -> f64

fn grad(&self, x: f64) -> f64

impl Transform<f64> for Exp

type Output = f64

fn transform(&self, x: f64) -> f64

fn grad(&self, x: f64) -> f64

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>> for Tanh

type Output = Vector<f64>

fn transform(&self, x: Vector<f64>) -> Vector<f64>

fn grad(&self, x: Vector<f64>) -> Vector<f64>

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>> for Identity

type Output = Vector<f64>

fn transform(&self, x: Vector<f64>) -> Vector<f64>

fn grad(&self, x: Vector<f64>) -> Vector<f64>

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>> for Softplus

type Output = Vector<f64>

fn transform(&self, x: Vector<f64>) -> Vector<f64>

fn grad(&self, x: Vector<f64>) -> Vector<f64>

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>> for LogSumExp

type Output = f64

fn transform(&self, x: Vector<f64>) -> f64

fn grad(&self, x: Vector<f64>) -> Vector<f64>

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>> for Logistic

type Output = Vector<f64>

fn transform(&self, x: Vector<f64>) -> Vector<f64>

fn grad(&self, x: Vector<f64>) -> Vector<f64>

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>> for Exp

type Output = Vector<f64>

fn transform(&self, x: Vector<f64>) -> Vector<f64>

fn grad(&self, x: Vector<f64>) -> Vector<f64>

Loading content...