Pytorch rolling window

import torch

x = torch.arange(10)#tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9])
x.unfold(0,4,3) #Tensor.unfold(dimension, window_size, step) 

'''tensor([[0, 1, 2, 3],
        [3, 4, 5, 6],
        [6, 7, 8, 9]])'''

Similar codes

2 votes314 views1 answer
Get dictionary in array python by value
Python
0 vote189 views1 answer
Numpy remove rows containing nan
Python
4 votes176 views1 answer
Record the amount of time ittales for code to run python
Python
1 vote144 views2 answers
Tkinter text color button
Python
18 votes196 views5 answers
Get list with random numbers python
Python
0 vote186 views1 answer
How to get data in treeview in tkiter
Python
0 vote236 views1 answer
Sin , Cos Graph using python turtle.
Python
96 votes238 views6 answers
Read json file
Python
1 vote231 views2 answers
Plt.scatter background color
Python
2 votes194 views2 answers
Python printing date
Python