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

23 votes302 views6 answers
Read pickle file python
Python
2 votes324 views1 answer
Python open url in incognito
Python
184 votes392 views6 answers
How to upgrade pip
Python
1 vote325 views1 answer
Python dictionary value as set
Python
0 vote315 views1 answer
Django raw without sql injection
Python
0 vote334 views1 answer
How to add phone number to django user model
Python Django
0 vote282 views1 answer
How to resume request downloads
Python
4 votes312 views2 answers
Django reset database
Python
0 vote296 views1 answer
If keyboard.is_pressed
Python
0 vote288 views1 answer
Custom permission class django rest framework
Python