Function dfdx::tensor_ops::powi

source · 
pub fn powi<S: Shape, E: Dtype, D: UnaryKernel<PowiKernelOp, E>, T: Tape<E, D>>(
    t: Tensor<S, E, D, T>,
    exponent: i32
) -> Tensor<S, E, D, T>
Expand description

Raises to an integer power; t^i.

let t = dev.tensor([-1.0, 0.0, 1.0, 2.0]);
let r = t.powi(3);