如何转换Matlab代码
v = [1: n]
到 torch ?
为此编写一个完整的循环似乎效率不高。
请您参考如下方法:
您可以直接使用 Pytorch 中的 arange 方法。
torch_v = torch.arange(1,n)
引用:https://pytorch.org/docs/master/torch.html?highlight=arange#torch.arange
如何转换Matlab代码
v = [1: n]
到 torch ?
为此编写一个完整的循环似乎效率不高。
请您参考如下方法:
您可以直接使用 Pytorch 中的 arange 方法。
torch_v = torch.arange(1,n)
引用:https://pytorch.org/docs/master/torch.html?highlight=arange#torch.arange