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: [list[int]|int, list[int]|int] | int

• shared: int | None

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

• :returns: _