AI engineering glossary · Linear algebra

What is tensor contraction?

Tensor contraction: Summing the products of entries of tensors over a shared axis, which reduces the total number of axes. Matrix multiplication, dot products and traces are all special cases.

Level: advancedTopic: Linear algebra

Tensor contraction: an example

Contracting a (2, 3) tensor with a (3, 4) tensor over the shared axis of size 3 yields a (2, 4) tensor.

Related terms

More Linear algebra terms

Where you learn it

Tensor contraction is part of the AI Engineering Program, which teaches it with notes, a hands-on lab and an exam. The first lesson is free.

Before you go

Take the cheat sheet with you

One free PDF: the key AI-engineering terms, formulas and Python idioms. No spam; unsubscribe any time.