Functions 5
block_diag
functionFull reference ↗- tensorplay._shape_funcs.block_diag(*tensors)[source]
Builds a block diagonal matrix from the given blocks.
0-D blocks become 1x1 matrices and 1-D blocks become diagonal
broadcast_shapes
functionFull reference ↗- tensorplay._shape_funcs.broadcast_shapes(*shapes)[source]
Returns the broadcast shape of the given shapes (right-aligned).
broadcast_tensors
functionFull reference ↗- tensorplay._shape_funcs.broadcast_tensors(*tensors)[source]
Broadcasts the given tensors to a common shape.
tensordot
functionFull reference ↗- tensorplay._shape_funcs.tensordot(input, other, dims=2)[source]
Contracts
inputandotherover the given dimensions.dimsmay be a non-negative int (contract the lastdimsdims ofinputwith the firstdimsdims ofother) or a pair(dims_a, dims_b)of dimension lists paired positionally.
unravel_index
functionFull reference ↗- tensorplay._shape_funcs.unravel_index(indices, shape)[source]
Converts flat indices into coordinate tuples (one LongTensor per dim).

