Fastest way to add matrices of different shapes in Python/Numba
Fastest way to add matrices of different shapes in Python/Numba Question: I want to "add" two matrices, a matrix a with shape (K,T) and a matrix b of shape (K,N), to result in a matrix of shape (K,T,N) The following works ok: import numpy as np from numba import njit @njit def add_matrices(a, b): K, …