Pytorch tensor教程练习

import torch as th
import numpy as np

data = [[1,2],[3,4]]
x_data = th.tensor(data)
print('x_data:',x_data)

np_array = np.array(data)
x_np = th.from_numpy(np_array)
print('x_np_from_numpy',x_np)

x_ones = th.ones_like(x_data)
print('x_ones_like',x_ones)

x_rand = th.rand_like(x_data,dtype=th.float)
print('x_rand_like',x_rand)
print('***********************************************************************')

shape = (2,3)
rand_tensor = th.rand(shape)
ones_tensor = th.ones(shape)
zero_tensor = th.zeros(shape)
print('rand_tensor:',rand_tensor)
print('one_tensor:',ones_tensor)
print('zeros_tensor:',zero_tensor)
print('***********************************************************************')

#Attributes tensor 张量属性
tensor = th.rand(3,3)
print('tensor:',tensor)
print('Shape of tensor:',tensor.shape)
print('Datatype of tensor:',tensor.dtype)
print('Device tensor is stored on :', tensor.device)
print('GPU can be use :',th.cuda.is_available())
#tensor = tensor.cuda()
tensor = tensor.to('cuda')
print('Device tensor is stored on:',tensor.device)
print('**********************Attributes tensor 张量属性************************')

#Operations on tensor 张量运算
tensor = th.ones(4,4)
print(tensor)
print('First row ',tensor[0])
print('First column ',tensor[:,0])
print('Last column ',tensor[:,-1])
tensor[:,1] = 0
print(tensor)
t1 = th.cat([tensor,tensor,tensor,tensor],dim=1)
print(t1)
print('#Operations on tensor 张量运算:矩阵乘法与矩阵转置相同

y1 = tensor@tensor.T
print('y1:',y1)
y2 = tensor.matmul(tensor.T)
print('y2:',y2)
y3 = th.rand_like(y1)
print('y3:',y3)
th.matmul(tensor,tensor.T,out=y3)
print('y3::',y3)
print('***#Operations on tensor 张量运算:矩阵点乘与mul同）******************')
z1 = tensor*tensor
print('z1:',z1)
z2 = tensor.mul(tensor)
print('z2',z2)
z3 = th.rand_like(tensor)
th.mul(tensor,tensor,out=z3)
print('z3:',z3)

print('**********#Operations on tensor 张量运算:单元素矩阵，通过item转python值）***')
agg = tensor.sum()
print(agg)
agg_item = agg.item()
print(agg_item,type(agg_item))
ay3 = y3.sum()
print(ay3)

print('tensor:',tensor)
tensor.add_(5)
print(tensor)
tensor1 = tensor.sub_(5)
print(tensor1)
print(tensor)
print('**********************************#使用Numpy桥接************')
t = th.ones(5)
print(t)
n = t.numpy()
print(n)
t.add_(1)
print(t)
print(n)
n = np.ones(10)
t = th.from_numpy(n)
print(t)
np.add(n,1,out=n)
print(n)
print(t)