noe icon indicating copy to clipboard operation
noe copied to clipboard

Implementing tensor dot

Open ariaghora opened this issue 5 years ago • 0 comments

Afaik, most of neural nets do not require tensor dot. However, wouldn't it be neat to have tensor dot implemented? First thing comes to my mind is treating tensor dot as batch matmul, so at least we can BLAS them.

Example of expected behavior: https://docs.scipy.org/doc/numpy-1.14.0/reference/generated/numpy.tensordot.html

ariaghora avatar Feb 17 '20 07:02 ariaghora