oa::matrix::scale

Multiply every FP32 matrix element by a scalar. The call records device work and returns without a host wait. Errors Returns an error when the shape, dtype, ownership, allocation, or recording contract fails.

FnRust public API

Function Documentation

pub fn scale(
	input: &Matrix,
	scalar: f32
) -> Result<Matrix>

Multiply every FP32 matrix element by a scalar. The call records device work and returns without a host wait. Errors Returns an error when the shape, dtype, ownership, allocation, or recording contract fails.