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

3 votes350 views2 answers
String of numbers to list of integers python
Python
6 votes342 views2 answers
How to add mouse button in pygame
Python
0 vote330 views1 answer
Why pytest return No ModuleError
Python
0 vote375 views1 answer
Ms access python dataframe
Python
0 vote327 views1 answer
Dockerize django app
Python
0 vote335 views1 answer
Python module to get football results
Python
1 vote312 views1 answer
Format listto string pytthon
Python
0 vote377 views1 answer
Math expressions with matplotlib
Python
27 votes343 views6 answers
How to check if there are duplicates in a list python
Python
0 vote383 views1 answer
Resize image array python
Python