makdoudN

'''
Created on 26.02.2016
Print numpy arrays to latex tables
@author: Peter Fischer
'''

from __future__ import print_function, division


def print_to_table(avgs, stds, rows, cols,

makdoudN

import numpy as np

from jax import numpy as jnp
from jax.lib import pytree


def reduce(trees, op=jnp.stack, op_kwargs: dict = {}):
    """Takes a list of trees and apply op to every corresponding leaf.
    For example, assuming op=`jnp.stack`,
    given two trees ((a, b), c) and ((a', b'), c'), returns