Baike.dev
All toolsAI codingTrendingOpen sourceNewsSubmit
Log in
Back to tool/Back to issues
#57·numpy-ml

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

View original on GitHubView discussion on GitHub