Skip to main content

gpu_log

ferrotorch_gpu::kernels

Function gpu_log 

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

Elementwise natural log: out[i] = ln(a[i]).