tensorplay._shape_funcs.tensordot

tensorplay._shape_funcs.tensordot(input, other, dims=2)[source]

Contracts input and other over the given dimensions.

dims may be a non-negative int (contract the last dims dims of input with the first dims dims of other) or a pair (dims_a, dims_b) of dimension lists paired positionally.