pytorch使用-tensor创建

pytorch使用-tensor创建

  • 一、导入包
  • 二、tensor创建方法
  • 三、tensor初始化和随机初始化
  • 四、tensor方法


一、导入包

import torch

二、tensor创建方法

pytorch使用-tensor创建_第1张图片

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)

pytorch使用-tensor创建_第2张图片

三、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)

pytorch使用-tensor创建_第3张图片

四、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())

pytorch使用-tensor创建_第4张图片

你可能感兴趣的:(pytorch,pytorch)