< Back

Env

import torch

\( \mathbf{A} \in \mathbb{R}^{m \times n} \)


  matrix

A = torch.arange(20).reshape(5, 4)
A 
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11],
        [12, 13, 14, 15],
        [16, 17, 18, 19]])