Skip to main content

Module utils

Module utils 

Source
Expand description

Utility functions for tensor operations

Functionsยง

are_broadcastable
Check if shapes are broadcastable
broadcast_shapes
Calculate output shape after broadcasting
matmul_output_shape
Calculate output shape for matrix multiplication