Type Definition aegir::ops::TensorDot

pub type TensorDot<L, R> = Contract<1, L, R>;

Operator alias that applies f[g,h](x) = h(x) · g(x) element-wise to a buffer.

Examples

let f = TensorDot::new(X.into_var(), Y.into_var());
let ctx = ctx!{
    X = [1.0, 2.0, 3.0],
    Y = [-1.0, 0.0, 2.0]
};

assert_eq!(f.evaluate_dual(X, &ctx).unwrap(), dual!(5.0, [-1.0, 0.0, 2.0]));
assert_eq!(f.evaluate_dual(Y, &ctx).unwrap(), dual!(5.0, [1.0, 2.0, 3.0]));

Trait Implementations

impl<L: ToExpr, R: ToExpr> Display for TensorDot<L, R>

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

Formats the value using the given formatter.

impl<L: ToExpr, R: ToExpr> ToExpr for TensorDot<L, R>

fn to_expr(&self) -> Expr

Convert the node to an expression string.