torch.linspace()

2022-09-02 22:07:45 浏览数 (1)

PyTorch中的torch.linspace

代码语言:javascript复制
 linspace是linear space的缩写,中文含义是线性等分向量

原函数

代码语言:javascript复制
torch.linspace(start, end, steps=100, out=None, dtype=None,layout=torch.strided, device=None, requires_grad=False)

参数

代码语言:javascript复制
start: 开始值
end:结束值
steps:分割的点数,默认为100

示例代码

代码语言:javascript复制
import torch
print(torch.linspace(1, 10, 5))

#输出结果:
    tensor([ 1.0000,  3.2500,  5.5000,  7.7500, 10.0000])

0 人点赞