View Source EXLA.Shape (EXLA v0.7.2)
Wrapper around XLA's shape.
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.
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.