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
GpuError::DeviceMismatchifaanddevicerefer to different CUDA devices.GpuError::Driveron CUDA runtime errors.