Pytorch中常见函数的使用

案例：

import torch

# 创建一个二维的Tensor
a = torch.Tensor([[1, 2], [3, 4]]) # 这个可以创建一个空的Tesor
print(a)
print(a.type())
a = torch.tensor([[1, 2], [3, 4]]) # 这个创建一个空的会报错
print(a)
print(a.type())

# 直接给形状
# 系统会随机分配值
a = torch.Tensor(2, 3)
print(a)
print(a.type())

# 创建一个元素为一的Tensor
a = torch.ones(size=[2, 2])
print(a)
print(a.type())

# 全零
a = torch.zeros(size=[2, 2])
print(a)
print(a.type())

# 对角线为1
a = torch.eye(2, 2)
print(a)
print(a.type())


# 可以根据定义的形状完成Tensor创建
a = torch.Tensor(2, 3)
b = torch.zeros_like(a) # 创建一个和a一样形状的Tensor值为1
print(b)
print(b.type())

a = torch.Tensor(2, 3)
b = torch.ones_like(a)
print(b)
print(b.type())


# 随机生成Tensor
a = torch.rand(2, 2)
print(a)
print(a.type())

# 随机生成正太分布Tensor
# a = torch.normal(mean=0.0, std=torch.rand(5))
a = torch.normal(mean=torch.rand(5), std=torch.rand(5))
print(a)
print(a.type())

# 随机生成均匀分布Tensor
a = torch.Tensor(2, 2).uniform_(-1, 1)
print(a)
print(a.type())


# 生成序列(前闭后开)
a = torch.arange(0,10,step=1)
print(a)
print(a.type())


# 等间隔切分
a = torch.linspace(1, 10, 5)
print(a)
print(a.type())

# 生成的随机的序列
a = torch.randperm(10)
print(a)
print(a.type())