【Pytorch基础】基础点整理1

# 创建tensor

a = np.array([2,2.3])
b = torch.from_numpy(a)
print(b)

tensor([2.0000, 2.3000], dtype=torch.float64)

c = np.ones([2,3])
d = torch.from_numpy(c)
print(d)

 tensor([[1., 1., 1.],
         [1., 1., 1.]], dtype=torch.float64)

# 从list导入

# 小写tensor放现成数据，大写Tensor放shape维度

a = torch.tensor([2.,3.2])
print(a)

tensor([2.0000, 3.2000])

# 大写也可以放现成数据但是必须得是个list，如果是维度的话只要（2，3）不需要[]

a = torch.FloatTensor([2.,3.2])
print(a)

tensor([2.0000, 3.2000])

# 定义空的容器准备放tensor（会生成随机数据，后续要输入）
# torch.empty(1)##如果不1或其他的话，会出现无穷数据
# torch.FloatTensor(1,2,3)##几维就几个数
# torch.IntTensor(1,2,3)

# 如果没有设置，不管是tensor还是Tensor都默认是float的

a = torch.tensor([1.2