使用pytorch生成形状为(100,)的以1开头顺序数组
时间: 2024-05-08 12:15:02 浏览: 90
以下是使用PyTorch生成形状为(100,)的以1开头顺序数组的示例代码:
```python
import torch
# 创建一个形状为(100,)的张量,元素全部为1
arr = torch.ones((100,))
# 生成以1开头的顺序数组
seq_arr = torch.cumsum(arr, dim=0)
print(seq_arr)
```
输出结果如下:
```
tensor([ 1., 2., 3., 4., 5., 6., 7., 8., 9., 10., 11., 12.,
13., 14., 15., 16., 17., 18., 19., 20., 21., 22., 23., 24.,
25., 26., 27., 28., 29., 30., 31., 32., 33., 34., 35., 36.,
37., 38., 39., 40., 41., 42., 43., 44., 45., 46., 47., 48.,
49., 50., 51., 52., 53., 54., 55., 56., 57., 58., 59., 60.,
61., 62., 63., 64., 65., 66., 67., 68., 69., 70., 71., 72.,
73., 74., 75., 76., 77., 78., 79., 80., 81., 82., 83., 84.,
85., 86., 87., 88., 89., 90., 91., 92., 93., 94., 95., 96.,
97., 98., 99., 100.])
```
可以看到,生成了一个形状为(100,)的张量,元素从1开始顺序递增。
阅读全文