Skip to main content
 首页 » 编程设计

python之如何在Pytorch中形成一个连续的数字序列

2025年05月04日9jiqing9006

如何转换Matlab代码

v = [1: n] 

到 torch ?

为此编写一个完整的循环似乎效率不高。

请您参考如下方法:

您可以直接使用 Pytorch 中的 arange 方法。

torch_v = torch.arange(1,n) 

引用:https://pytorch.org/docs/master/torch.html?highlight=arange#torch.arange