Introduction to Matrices and Matrix Arithmetic for Machine Learning

Matrices are a foundational element of linear algebra.

Matrices are used throughout the field of machine learning in the description of algorithms and processes such as the input data variable (X) when training an algorithm.

In this tutorial, you will discover matrices in linear algebra and how to manipulate them in Python.

After completing this tutorial, you will know:

  • What a matrix is and how to define one in Python with NumPy.
  • How to perform element-wise operations such as addition, subtraction, and the Hadamard product.
  • How to multiply matrices together and the intuition behind the operation.

Kick-start your project with my new book Linear Algebra for Machine Learning, including step-by-step tutorials and the Python source code files for all examples.

Let’s get started.

  • Update Jun/2019: Fixed a typo in the Matrix-Vector Multiplication section (thanks M. Vincent).
  • Update Jun/2019: Fixed typo in description of matrix in Python (thanks Ari).
  • Update Oct/2021: Added example of using @-operator for matrix-vector multiplication.
A Gentle Introduction to Matrices for Machine Learning

A Gentle Introduction to Matrices for Machine Learning
Photo by Maximiliano Kolus, some rights reserved.

Tutorial Overview

This tutorial is divided into 6 parts; they are:

  1. What is a Matrix?
  2. Defining a Matrix
  3. Matrix Arithmetic
  4. Matrix-Matrix Multiplication (Dot Product)
  5. Matrix-Vector Multiplication
  6. Matrix-Scalar Multiplication

Need help with Linear Algebra for Machine Learning?

Take my free 7-day email crash course now (with sample code).

Click to sign-up and also get a free PDF Ebook version of the course.

What is a Matrix?

A matrix is a two-dimensional array of scalars with one or more columns and one or more rows.

A matrix is a two-dimensional array (a table) of numbers.

— Page 115, No Bullshit Guide To Linear Algebra, 2017

The notation for a matrix is often an uppercase letter, such as A, and entries are referred to by their two-dimensional subscript of row (i) and column (j), such as aij. For example:

It is more common to see matrices defined using a horizontal notation.

A likely first place you may encounter a matrix in machine learning is in model training data comprised of many rows and columns and often represented using the capital letter “X”.

The geometric analogy used to help understand vectors and some of their operations does not hold with matrices. Further, a vector itself may be considered a matrix with one column and multiple rows.

Often the dimensions of the matrix are denoted as m and n for the number of rows and the number of columns.

Now that we know what a matrix is, let’s look at defining one in Python.

Defining a Matrix

We can represent a matrix in Python using a two-dimensional NumPy array.

A NumPy array can be constructed given a list of lists. For example, below is a 2 row, 3 column matrix.

Running the example prints the created matrix showing the expected structure.

Matrix Arithmetic

In this section will demonstrate simple matrix-matrix arithmetic, where all operations are performed element-wise between two matrices of equal size to result in a new matrix with the same size.

Matrix Addition

Two matrices with the same dimensions can be added together to create a new third matrix.

The scalar elements in the resulting matrix are calculated as the addition of the elements in each of the matrices being added.

Or, put another way:

We can implement this in python using the plus operator directly on the two NumPy arrays.

The example first defines two 2×3 matrices and then adds them together.

Running the example first prints the two parent matrices and then the result of adding them together.

Matrix Subtraction

Similarly, one matrix can be subtracted from another matrix with the same dimensions.

The scalar elements in the resulting matrix are calculated as the subtraction of the elements in each of the matrices.

Or, put another way:

We can implement this in python using the minus operator directly on the two NumPy arrays.

The example first defines two 2×3 matrices and then subtracts one from the other.

Running the example first prints the two parent matrices and then subtracts the first matrix from the second.

Matrix Multiplication (Hadamard Product)

Two matrices with the same size can be multiplied together, and this is often called element-wise matrix multiplication or the Hadamard product.

It is not the typical operation meant when referring to matrix multiplication, therefore a different operator is often used, such as a circle “o”.

As with element-wise subtraction and addition, element-wise multiplication involves the multiplication of elements from each parent matrix to calculate the values in the new matrix.

Or, put another way:

We can implement this in python using the star operator directly on the two NumPy arrays.

The example first defines two 2×3 matrices and then multiplies them together.

Running the example first prints the two parent matrices and then the result of multiplying them together with a Hadamard Product.

Matrix Division

One matrix can be divided by another matrix with the same dimensions.

The scalar elements in the resulting matrix are calculated as the division of the elements in each of the matrices.

Or, put another way:

We can implement this in python using the division operator directly on the two NumPy arrays.

The example first defines two 2×3 matrices and then divides the first from the second matrix.

Running the example first prints the two parent matrices and then divides the first matrix by the second.

Matrix-Matrix Multiplication (Dot Product)

Matrix multiplication, also called the matrix dot product is more complicated than the previous operations and involves a rule as not all matrices can be multiplied together.

or

The rule for matrix multiplication is as follows:

  • The number of columns (n) in the first matrix (A) must equal the number of rows (m) in the second matrix (B).

For example, matrix A has the dimensions m rows and n columns and matrix B has the dimensions n and k. The n columns in A and n rows b are equal. The result is a new matrix with m rows and k columns.

This rule applies for a chain of matrix multiplications where the number of columns in one matrix in the chain must match the number of rows in the following matrix in the chain.

One of the most important operations involving matrices is multiplication of two matrices. The matrix product of matrices A and B is a third matrix C. In order for this product to be defined, A must have the same number of columns as B has rows. If A is of shape m × n and B is of shape n × p, then C is of shape m × p.

— Page 34, Deep Learning, 2016.

The intuition for the matrix multiplication is that we are calculating the dot product between each row in matrix A with each column in matrix B. For example, we can step down rows of column A and multiply each with column 1 in B to give the scalar values in column 1 of C.

This is made clear with the following image.

Depiction of matrix multiplication.

Depiction of matrix multiplication, taken from Wikipedia, some rights reserved.

Below describes the matrix multiplication using matrix notation.

This can be simplified by removing the multiplication signs as:

We can describe the matrix multiplication operation using array notation.

The matrix multiplication operation can be implemented in NumPy using the dot() function.

The example first defines two 2×3 matrices and then calculates their dot product.

Running the example first prints the two parent matrices and then the result of the dot product.

Matrix-Vector Multiplication

A matrix and a vector can be multiplied together as long as the rule of matrix multiplication is observed.

Specifically, that the number of columns in the matrix must equal the number of items in the vector. As with matrix multiplication, the operation can be written using the dot notation. Because the vector only has one column, the result is always a vector.

Or without the dot in a compact form.

The result is a vector with the same number of rows as the parent matrix.

Or, more compactly.

The matrix-vector multiplication can be implemented in NumPy using the dot() function or the @ operator (since Python version 3.5).

The example first defines a 3×2 matrix and a 2 element vector and then multiplies them together.

Running the example first prints the parent matrix and vector and then the result of multiplying them together.

Matrix-Scalar Multiplication

A matrix can be multiplied by a scalar.

This can be represented using the dot notation between the matrix and the scalar.

Or without the dot notation.

The result is a matrix with the same size as the parent matrix where each element of the matrix is multiplied by the scalar value.

or

We can also represent this with array notation.

This can be implemented directly in NumPy with the multiplication operator.

The example first defines a 2×3 matrix and a scalar and then multiplies them together.

Running the example first prints the parent matrix and scalar and then the result of multiplying them together.

Extensions

This section lists some ideas for extending the tutorial that you may wish to explore.

  • Create 5 examples using each operation using your own data.
  • Implement each matrix operation manually for matrices defined as lists of lists.
  • Search machine learning papers and find 1 example of each operation being used.

If you explore any of these extensions, I’d love to know.

More Tutorials

There are many different types of matrices, for some examples see:

Matrices are used in many different operations, for some examples see:

Further Reading

This section provides more resources on the topic if you are looking to go deeper.

Books

API

Articles

Summary

In this tutorial, you discovered matrices in linear algebra and how to manipulate them in Python.

Specifically, you learned:

  • What a matrix is and how to define one in Python with NumPy.
  • How to perform element-wise operations such as addition, subtraction, and the Hadamard product.
  • How to multiply matrices together and the intuition behind the operation.

Do you have any questions?
Ask your questions in the comments below and I will do my best to answer.

Get a Handle on Linear Algebra for Machine Learning!

Linear Algebra for Machine Learning

Develop a working understand of linear algebra

...by writing lines of code in python

Discover how in my new Ebook:
Linear Algebra for Machine Learning

It provides self-study tutorials on topics like:
Vector Norms, Matrix Multiplication, Tensors, Eigendecomposition, SVD, PCA and much more...

Finally Understand the Mathematics of Data

Skip the Academics. Just Results.

See What's Inside

13 Responses to Introduction to Matrices and Matrix Arithmetic for Machine Learning

  1. Avatar
    Erick February 7, 2018 at 8:16 pm #

    If someone wants to *understand* what matrices actually do instead of applying them in Python, I truly recommend watching the YouTube channel “Linear Algebra” by 3Blue1Brown. Very insightful, requires very minimal knowledge of vectors and matrices.

    • Avatar
      Jason Brownlee February 8, 2018 at 8:24 am #

      Thanks Erick.

    • Avatar
      Russell February 17, 2018 at 10:07 am #

      Thanks Erick. Great resource!!

    • Avatar
      samir khan July 19, 2023 at 10:11 pm #

      thanks Erick

  2. Avatar
    Russell Bigley February 16, 2018 at 12:26 pm #

    I think there is an error in the code. The Matrix vector Multiplication

    should be

  3. Avatar
    meee June 7, 2018 at 9:16 am #

    Hi sir can you please guide me. i want to learn absolute algebra. I will be waiting for your reply. Thanks a lot.

  4. Avatar
    M. Vincent June 11, 2019 at 8:46 pm #

    In the Matrix-Vector multiplication section you state:

    “The example first defines a 2×3 matrix and a 2 element vector and then multiplies them together.”

    Your example has a 3×2 matrix, and a 2 element row vector.

  5. Avatar
    Ari June 26, 2019 at 11:34 am #

    In your very first example you state:

    A NumPy array can be constructed given a list of lists. For example, below is a 3 row, 2 column matrix.

    # create matrix
    from numpy import array
    A = array([[1, 2, 3], [4, 5, 6]])
    print(A)
    Running the example prints the created matrix showing the expected structure.

    [[1 2 3]
    [4 5 6]]

    This a 2 row, 3 column matrix — correct?

    • Avatar
      Jason Brownlee June 26, 2019 at 2:37 pm #

      Correct, I have updated the example. Thanks.

      • Avatar
        Ari June 28, 2019 at 11:51 am #

        Cheers!

Leave a Reply