module testing.experimental#

Short summary#

module mlprodict.testing.experimental

Experimental implementation.

source on GitHub

Functions#

function

truncated documentation

custom_einsum

Experimental implementation of operator Einsum when it does a matrix multiplication. Case: bsnh,btnh->bnts with …

custom_pad

Implements function pad in python, only …

Documentation#

Experimental implementation.

source on GitHub

mlprodict.testing.experimental.custom_einsum(equation, x, y, verbose=False)#

Experimental implementation of operator Einsum when it does a matrix multiplication. Case: bsnh,btnh->bnts with shapes (1,512,12,64) and (1,512,12,64).

Parameters:

  • equation – equation

  • x – first matrix

  • y – second matrix

  • verbose – display internal information

Returns:

result of einsum

This implementation does not any transpose, it does a direct computation of the final result. It does not implementation diagonal summation (square product).

source on GitHub

mlprodict.testing.experimental.custom_pad(arr, paddings, constant=0, verbose=False)#

Implements function pad in python, only the constant version.

Parameters:

  • arr – array

  • paddings – paddings

  • constant – constant

Returns:

padded array

source on GitHub