pytorch.range() 和 pytorch.arange()
pytorch.range() 和 pytorch.arange()
狼啸风云
修改于 2022-09-02 22:33:50
修改于 2022-09-02 22:33:50
文章被收录于专栏：计算机视觉理论及其实现
>>> 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/04/06 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
