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

0 vote305 views3 answers
Python compiler to c++
Python
1 vote322 views1 answer
Match python 3.10
Python
0 vote295 views1 answer
Pss signatures python
Python
5 votes322 views2 answers
Selenium get current url
Python
1 vote328 views1 answer
Tqdm parallel
Python
3 votes291 views1 answer
Create a window turtle python
Python
0 vote320 views1 answer
Progress apply
Python
0 vote341 views1 answer
Python fibonacci sequence for loop
Python
0 vote339 views2 answers
Displaying flash message django
Python Django
0 vote318 views1 answer
Live server python
Python