Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
4 years ago
import torch 
import numpy as np
t1 = torch.tensor(4.0)

t1.dtype
torch.float32
vec = torch.tensor([1,2,3,4])
vec
tensor([1, 2, 3, 4])
t3 = torch.tensor([[5., 6], 
                [7, 8], 
                [9, 10]])
t3.shape
torch.Size([3, 2])
# 3D array
t4 = torch.tensor([
    [[11, 12, 13], 
     [13, 14, 15]], 
    [[15, 16, 17], 
     [17, 18, 19.]]])
t4
tensor([[[11., 12., 13.],
         [13., 14., 15.]],

        [[15., 16., 17.],
         [17., 18., 19.]]])