torch.linspace

原创已于 2022-09-07 10:42:28 修改 · 696 阅读

CC 4.0 BY-SA版权

文章标签：

于 2022-09-07 10:39:13 首次发布

33 篇文章

订阅专栏

本文解析了torch.linspace函数中第三个参数的实际含义，强调它代表被采样的元素数目，通过两个实例对比说明其与range函数的区别。

注意，torch.linsapce 第三个参数的含义与 range函数第三个参数的含义不同。

例子1

import torch

print(torch.linspace(-5,5,11))

注意，第三个参数的含义是
被采样的元素的数目
在这里的例子中是 11 不是 10！

输出：

tensor([-5., -4., -3., -2., -1.,  0.,  1.,  2.,  3.,  4.,  5.])

import torch


print(torch.linspace(-5,5,10))

当取10 时

输出

tensor([-5.0000, -3.8889, -2.7778, -1.6667, -0.5556,  0.5556,  1.6667,  2.7778,
         3.8889,  5.0000])