asc.language.tile.binary_ops.matmul

asc.language.tile.binary_ops.matmul(input: Tile, other: Tile) → Tile

Computes the matrix multiplication of input and other.

Parameters:

• input – the left operand (2D tile)

• other – the right operand (2D tile)

Returns:

the result of the matrix multiplication

Return type:

Tile

Note

Input tiles must have either float16 or float32 data type and compatible shapes. Result tile type is always float32.

This function can also be called as a member function on Tile, as x.__matmul__(...) instead of matmul(x, ...).