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

34 votes150 views1 answer
Python main file with class
Python
0 vote159 views1 answer
Python verificar se é numero
Python
0 vote125 views1 answer
Python move all txt files
Python
30 votes234 views3 answers
Django createsuperuser
Python Django
0 vote162 views1 answer
Check this id exist in database django
Python
1 vote163 views1 answer
How to create an array with known length in python
Python
1 vote177 views2 answers
Mp4 to mp3 in python
Python
0 vote225 views1 answer
How to deploy to heroku djangorest framework app
Python Django
0 vote243 views1 answer
Queryset.raw() in django rest framework joining tables
Python Django
1 vote175 views1 answer
Python retry
Python