check pytorch tensor size

check pytorch tensor size

 

import torch

## allocate sample tensor to x
x = torch.FloatTensor([
    [1, 2],
    [3, 4],
    [5, 6],
    [7, 8]
])

## size with "size" method
x.size()
# out : torch.Size([4, 2])

## size with "shape" method
x.shape
# out : torch.Size([4, 2])

## dimension with "dim" method
x.dim()
# out : 2

## size on 0 dimension with "size(0)" method
x.size(0)
# out : 4

x.shape[0]
# out: 4