Torch.arange函数的使用

torch.arange的官方文档为:点这里
torch.arange的用法简述:返回大小为 (end-start)/step的一维张量,值在[strat,end),以step为步长等间隔取值,输入参数分别为strat,end,step。
torch.arange函数实现的效果为:

x_test = torch.arange(0, 3, 0.5)

输出x_test效果为:

x_test= tensor([0.0000, 0.5000, 1.0000, 1.5000, 2.0000, 2.5000])

值得注意的是strat的默认值为0,step的默认值为1

你可能感兴趣的:(深度学习,python,pytorch)