Pytorch 学习笔记--torch.linspace()函数

先看Pytorch官方的解释:

Pytorch 学习笔记--torch.linspace()函数_第1张图片

 简单来说,就是返回一个一维张量,这个张量包含由start到end的均匀间隔steps的n个点

参数:

start (float) - 区间的起始点

end (float) - 区间的终点

steps (int) - 在start和end间生成的样本数

out (Tensor, optional) - 结果张量

例子:

a = torch.linspace(0,10,2)
print(a)
# tensor([0.0000, 2.0000, 4.0000, 6.0000, 8.0000, 10.0000])

你可能感兴趣的:(pytorch学习笔记,python学习笔记,pytorch,深度学习,人工智能)