Python的PyTorch库技术点案例示例:深度解读PyTorch深度学习在医学领域应用
import torch
# 创建一个形状为(2, 3)的Tensor,初始化为0
x = torch.zeros(2, 3)
print(x)
# 创建一个形状为(2, 3)的Tensor,初始化为1
y = torch.ones(2, 3)
print(y)
# 创建一个形状为(2, 3)的Tensor,使用正态分布随机初始化
z = torch.randn(2, 3)
print(z)
# 创建一个形状为(2, 3)的Tensor,初始化为特定值
w = torch.full((2, 3), 3)
print(w)
# 创建一个形状为(2, 3)的Tensor,使用arange创建一个序列
a = torch.arange(0, 6).view(2, 3)
print(a)
# 创建一个形状为(2, 3)的Tensor,使用线性间隔创建一个序列
b = torch.linspace(0, 5, 6).view(2, 3)
print(b)
# 创建一个形状为(2, 3)的Tensor,使用特定的序列进行初始化
c = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(c)
这段代码展示了如何使用PyTorch创建不同类型的Tensor对象,包括零tensor、一tensor、正态分布随机tensor、特定值tensor、序列tensor以及通过已有数据创建的tensor。这些创建方法对于初学者理解PyTorch中tensor的创建和初始化非常有帮助。
评论已关闭