vec_tensordot

vec_tensordot(tensa, tensb, axes=2, shared=None): 

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

• tensb: Any

• axes: Any

• shared: int | None

  the axes that should be treated as shared (for now just an int)

• :returns: _