torch模块

import torch 
# 创建一个tensor对象
x = torch.empty(5,3)
print(x)

x = torch.rand(5,3)
print(x)

x = torch.zeros(5,3)
print(x)

x = torch.ones(5,3)
print(x)

print(x.dtype)
print(type(x))

list = [1,2,3]
x = torch.tensor(list)
print(x)

x.size() #返回的是一个tuple 

# 运算函数
y = torch.rand(5,3)
c = torch.add(x,y)

# 切片操作 使用[ ]
z = x[1：，1：]
print(z)

# 形状操作 一定要可以整除
w = x.view(15)
print(w)

# 类型转换操作
torch.from_numpy()
a = x.item()
b = x.numpy()