Skip to main content

gpu_scale

Function gpu_scale 

Source
pub fn gpu_scale(
    a: &CudaBuffer<f32>,
    scalar: f32,
    device: &GpuDevice,
) -> GpuResult<CudaBuffer<f32>>
Expand description

Scalar multiply: out[i] = a[i] * scalar.

Multiplies every element by a constant float value on the GPU.

ยงErrors