pytorch使用-tensor创建
一、导入包
import torch
二、tensor创建方法
x = torch.empty(2, 3)
print(x)
x = torch.Tensor(2, 3)
print(x)
x = torch.zeros(2, 3, dtype=torch.long)
print(x)
x = torch.ones(2, 3, dtype=torch.long)
print(x)
三、tensor初始化和随机初始化
x = torch.tensor([[1, 2, 6], [3, 4, 6]])
print(x)
# 是[0,1)之间的均匀分布
x = torch.rand(2, 3)
print(x)
# 标准正态分布中抽取的一组随机数的张量
x = torch.randn(2, 3)
print(x)
四、tensor方法
# 标准正态分布中抽取的一组随机数的张量
x = torch.randn(2, 3)
print(x)
# 改变x的形状和值
x = x.new_ones(4, 3)
print(x)
# 创建和x同形状的tensor
y = torch.randn_like(x, dtype=torch.float32)
print(y)
print(y.size())