Function tch_utils::ops_2d::scale_2d

pub fn scale_2d(t: &Tensor, scale: &[f64]) -> Tensor

Scale the of vectors in 2d pictures.

Arguments

t: Tensor - The tensor to scale [N, C, H, W] scale: &f64 - The scale factors (lenght == C)

Returns

Tensor - The scaled tensor [N, C, H, W]