torch.range()和torch.arange()的区别
狼啸风云
修改于 2022-09-02 13:50:19
修改于 2022-09-02 13:50:19
代码可运行
运行总次数：0
代码可运行
关联问题
换一批
>>> y=torch.range(1,6)
>>> y
tensor([1., 2., 3., 4., 5., 6.])
>>> y.dtype
torch.float32
>>> z=torch.arange(1,6)
>>> z
tensor([1, 2, 3, 4, 5])
>>> z.dtype
torch.int64总结：
torch.range(start=1, end=6)的结果是会包含end的， 而torch.arange(start=1, end=6)的结果并不包含end。- 两者创建的
tensor的类型也不一样。
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2020/03/20 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
暂无评论
推荐阅读
腾讯云开发者
