torch.linspace()的用法

最新推荐文章于 2026-01-29 08:26:13 发布

原创最新推荐文章于 2026-01-29 08:26:13 发布 · 516 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

pytorch深度学习

当前文章被收录于：

pytorch深度学习

5 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

torch.linspace()是一个用于创建等差序列的函数，它接受开始值、结束值和步长作为参数，返回一个Tensor。例如，从3到10，分5步创建的序列是[3.0000,4.7500,6.5000,8.2500,10.0000]，可以指定数据类型如float。此函数在数值计算和深度学习模型中常见，类似于Numpy的linspace功能。

torch.linspace()的用法

语法：
torch.linspace(start, end, steps, dtype)

参数：
start: 开始值
end: 结束值
steps: 分割点数，默认是100
dtype: 返回值（tensor）的数据类型

示例：

>>> import torch
>>> print(torch.linspace(3,10,5))
tensor([ 3.0000, 4.7500, 6.5000, 8.2500, 10.0000])

>>> type=torch.float
>>> print(torch.linspace(-10,10,steps=6,dtype=type))
tensor([-10., -6., -2., 2., 6., 10.])