API symbols_shape_funcs
Copy
tensorplay._shape_funcs.block_diag
- 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
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?
tensorplay._einsum.einsum
Sums the product of the elements of the input operands along dimensions specified using a notation based on the Einstein summation convention, e.g. tp.einsum("ij,jk->ik", A, B) computes a matrix multiplication.
tensorplay._shape_funcs.broadcast_shapes
Returns the broadcast shape of the given shapes (right-aligned).