AI智能
改变未来

啁啾 (@Chirp) 在 torch 2.9.1 我傻了 中发帖 把 torch 升到 2.9.1 发现原来的扩散模型跑不通了。然后发现了一个很离谱的问题。 torch.linspace(…

啁啾 (@Chirp) 在 torch 2.9.1 我傻了 中发帖

把 torch 升到 2.9.1 发现原来的扩散模型跑不通了。然后发现了一个很离谱的问题。
torch.linspace(0, 999, 20).long() 得到的是
tensor([ 0, 52, 105, 158, 211, 264, 316, 370, 422,
474, 528, 576, 632, 684, 736, 788, 840, 896, 948, 1000])

怎么超范围了,这个 1000 是啥
前段时间的 sdpa 运算问题也是,感觉 torch 2.9 不是很稳定

详情链接:
https://linux.do/t/topic/1345127/1

来源: LINUX DO, 消息ID: 263870

赞(0) 打赏
未经允许不得转载:爱站程序员基地 » 啁啾 (@Chirp) 在 torch 2.9.1 我傻了 中发帖 把 torch 升到 2.9.1 发现原来的扩散模型跑不通了。然后发现了一个很离谱的问题。 torch.linspace(…