View Source EXLA.Shape (EXLA v0.4.1)
Wrapper around XLA's shape.
Link to this section Summary
Functions
Returns the shape size in bytes.
Converts a charlist type into Nx' tuple format.
Converts Nx's tuple format into charlist.
Creates a shape with the given type-size tuple and dimensions.
Creates a token shape.
Creates a tuple shape with the given shapes.
Link to this section Functions
Returns the shape size in bytes.
Converts a charlist type into Nx' tuple format.
Converts Nx's tuple format into charlist.
Creates a shape with the given type-size tuple and dimensions.
Creates a token shape.
Creates a tuple shape with the given shapes.