Using numpy.tensordot for Conv2D
Author: tetrahydraCreated Aug 2, 2020Updated Aug 20, 2020
From this link: https://stackoverflow.com/questions/56085669/convolutional-layer-in-python-using-numpy
and
https://numpy.org/doc/stable/reference/generated/numpy.tensordot.html
Z = np.tensordot(X_pad, weights, axes=3) + self.bias
Does this function is more relevant that using im2col?
Source: ddbourgin/numpy-ml