TensorPlay
API symbols_shape_funcs
Copy
View MarkdownDownload .md

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.

tensorplay._shape_funcs.broadcast_shapes

Returns the broadcast shape of the given shapes (right-aligned).

tensorplay._shape_funcs.unravel_index

Converts flat indices into coordinate tuples (one LongTensor per dim).

Ask DeepWiki