pytorch torch.arange函数介绍
torch.arange 是 PyTorch 中用于创建具有等间距值的一维张量的函数。它类似于 Python 的 range 函数，但生成的是张量。
函数定义
torch.arange(start=0, end, step=1, *, dtype=None, layout=torch.strided, device=None, requires_grad=False)
参数说明
-
start(float 或 int, 可选，默认值：0):- 起始值（包含）。
-
end(float 或 int):- 结束值（不包含）。
-
step(float 或 int, 可选，默认值：1):- 步长。指定每两个相邻值之间的间隔，可以为负数。
-
dtype(torch.dtype, 可选):- 张量的数据类型。如果未指定，将根据
start和end推断。
- 张量的数据类型。如果未指定，将根据
-
layout(torch.layout, 可选):