vec_tensordot
vec_tensordot(tensa, tensb, axes=2):
Defines a version of tensordot that uses matmul to operate over stacks of things Basically had to duplicate the code for regular tensordot but then change the final call
tensa
:Any
No description…
tensb
:Any
No description…
axes
:Any
No description…
:returns
:_
No description…
Examples:
Edit Examples or
Create New Examples
Edit Template or
Create New Template
Edit Docstrings