tensorplay.py

#best part with pytorch is you can treat pytorch object as python object 
var=torch.rand(5,2,dtype=torch.double)

#we can loop over it
for i in var:
    for k in i:
        print(k)
        
# output:
# tensor(0.9142, dtype=torch.float64)
# tensor(0.9585, dtype=torch.float64)
# tensor(0.5760, dtype=torch.float64)
# tensor(0.2111, dtype=torch.float64)
# tensor(0.4621, dtype=torch.float64)
# tensor(0.4718, dtype=torch.float64)
# tensor(0.8199, dtype=torch.float64)
# tensor(1.00000e-02 *
#        8.4888, dtype=torch.float64)
# tensor(1.00000e-02 *
#        7.0895, dtype=torch.float64)
# tensor(0.9623, dtype=torch.float64)




#we can slice it ?
print(var[:2])

# output:
  
# tensor([[ 0.9142,  0.9585],
#         [ 0.5760,  0.2111]], dtype=torch.float64)


#we can play with it
print(list(map(lambda x:x[0],var)))


# output:
# [tensor(0.9142, dtype=torch.float64), tensor(0.5760, dtype=torch.float64), tensor(0.4621, dtype=torch.float64), tensor(0.8199, dtype=torch.float64), tensor(1.00000e-02 *
#        7.0895, dtype=torch.float64)]