In [1]:
import torch
In [2]:
one_d = torch.arange(2, 7)
one_d
Out[2]:
tensor([2, 3, 4, 5, 6])
In [3]:
one_d = torch.arange(2, 7, 2)
one_d
Out[3]:
tensor([2, 4, 6])
In [4]:
one_d = torch.arange(0, 9)
one_d
Out[4]:
tensor([0, 1, 2, 3, 4, 5, 6, 7, 8])
In [5]:
two_d = one_d.view(3, 3)
two_d
Out[5]:
tensor([[0, 1, 2],
        [3, 4, 5],
        [6, 7, 8]])
In [6]:
two_d.dim() # the number of dimentions
Out[6]:
2
In [7]:
two_d[0, 2]
Out[7]:
tensor(2)
In [8]:
two_d[2, 2]
Out[8]:
tensor(8)
In [9]:
x = torch.arange(18).view(2, 3, 3)
x
Out[9]:
tensor([[[ 0,  1,  2],
         [ 3,  4,  5],
         [ 6,  7,  8]],

        [[ 9, 10, 11],
         [12, 13, 14],
         [15, 16, 17]]])
In [10]:
x = torch.arange(18).view(3, 3, 2)
x
Out[10]:
tensor([[[ 0,  1],
         [ 2,  3],
         [ 4,  5]],

        [[ 6,  7],
         [ 8,  9],
         [10, 11]],

        [[12, 13],
         [14, 15],
         [16, 17]]])