Module `ambit_stochastics.helpers.alternative_convolution_implementation`

Expand source code

import numpy as np
def cumulative_and_diagonal_sums(a):
    k = a.shape[-1] #nr_trawls
    a = np.cumsum(a[::-1],axis=0)[::-1]
    return np.bincount(sum(np.indices(a.shape)).flat, a.flat)[:k]

Functions

def cumulative_and_diagonal_sums(a)

Expand source code

def cumulative_and_diagonal_sums(a):
    k = a.shape[-1] #nr_trawls
    a = np.cumsum(a[::-1],axis=0)[::-1]
    return np.bincount(sum(np.indices(a.shape)).flat, a.flat)[:k]