TensorPlay

Latest development documentation · Updated 2026-09-08. A documentation snapshot for package 1.0.0.dev20260909 is not available.

On this page

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 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.

#

unravel_index

functionFull reference ↗
tensorplay._shape_funcs.unravel_index(indices, shape)[source]

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

Search documentation

Search all 1,743 documentation pages.

Keyboard shortcuts

Global

  • /Focus search
  • ?This dialog
  • ,Open settings
  • jAI assistant

Search

  • Navigate results
  • Open result
  • escClose

Package

  • mMain information
  • dDocs
  • .Code
  • -Changelog
  • tTimeline
  • sStats
  • vVersions