Function dfdx::tensor_ops::axpy

source ·

pub fn axpy<S: Shape, E: Dtype, D>(
    a: &Tensor<S, E, D>,
    alpha: impl Into<f64>,
    b: &Tensor<S, E, D>,
    beta: impl Into<f64>
) -> Tensor<S, E, D>where
    D: AxpyKernel<E>,

Expand description

Elementwise a * alpha + b * beta.

See Tensor::axpy for in place version.