GpuForecasting

Trait GpuForecasting 

Source
pub trait GpuForecasting<F: Float + Debug> {
    // Required methods
    fn forecast_gpu(
        &self,
        steps: usize,
        config: &GpuConfig,
    ) -> Result<Array1<F>>;
    fn batch_forecast_gpu(
        &self,
        data: &[Array1<F>],
        steps: usize,
        config: &GpuConfig,
    ) -> Result<Vec<Array1<F>>>;
}
Expand description

GPU-accelerated forecasting operations

Required Methods§

Source

fn forecast_gpu(&self, steps: usize, config: &GpuConfig) -> Result<Array1<F>>

Perform forecasting on GPU

Source

fn batch_forecast_gpu( &self, data: &[Array1<F>], steps: usize, config: &GpuConfig, ) -> Result<Vec<Array1<F>>>

Batch forecasting for multiple series

Implementors§