ChainMatmul

Performs matrix multiplication of N tensors.

Abstract Signature:

ChainMatmul(matrices: List[Tensor])

PyTorch

API: torch.chain_matmul
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.linalg.multi_dot
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.linalg.multi_dot
Strategy: Direct Mapping
Official Docs ↗