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

4 votes193 views4 answers
Python save output to file
Python
0 vote219 views1 answer
Identifying strings python
Python
11 votes196 views2 answers
Regression python
Python
0 vote219 views2 answers
Install pyttsx3
Python
0 vote192 views1 answer
Difference between calling a function and referencing a function python
Python
3 votes209 views1 answer
Import py to exe
Python
0 vote200 views1 answer
Size.width>0 && size.height>0 in function 'cv::imshow'
Python
2 votes183 views1 answer
Pandas series select first value
Python
12 votes227 views5 answers
Python inf
Python
18 votes178 views9 answers
Python sort dictionary by value
Python