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.

Link to this function

make_tuple_shape(shapes)

View Source

Creates a tuple shape with the given shapes.