DiagEmbed¶
Creates a tensor whose diagonals of certain 2D planes are filled by input.
Abstract Signature:
DiagEmbed(input: Tensor, offset: int = 0, dim1: int = -2, dim2: int = -1)
JAX (Core)
API:
—Strategy: Custom / Partial
Creates a tensor whose diagonals of certain 2D planes are filled by input.
Abstract Signature:
DiagEmbed(input: Tensor, offset: int = 0, dim1: int = -2, dim2: int = -1)
—