torch.linspace
是 PyTorch 提供的一个用于生成等间隔数值的函数。具体而言,torch.linspace
会在指定的区间内生成指定数量的等间隔数值。
torch.linspace(start, end, steps=100, dtype=None, layout=torch.strided, device=None, requires_grad=False)
参数说明:
start
: 序列的起始值。end
: 序列的结束值。steps
: 生成的数值数量,默认为 100。dtype
: 生成的张量的数据类型,默认为None
,表示采用默认的数据类型。layout
: 生成的张量的布局,默认为torch.strided
。device
: 生成的张量所在的设备,默认为None
,表示使用默认设备。requires_grad
: 是否需要梯度,默认为False
。
示例
import torch# 生成一个包含 5 个等间隔数值的张量,范围从 1 到 2
values = torch.linspace(1, 2, 5)
print(values)
在这个例子中,torch.linspace(1, 2, 5)
将生成一个包含 5 个等间隔数值的张量,范围从 1 到 2。输出可能类似于:
tensor([1.0000, 1.2500, 1.5000, 1.7500, 2.0000])
这些数值是在指定范围内均匀分布的。