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 vote219 views1 answer
Pydub create empty track
Python
0 vote243 views1 answer
Rectangle function numpy
Python
0 vote216 views1 answer
How to call object of a list python
Python
3 votes208 views1 answer
Python array delete last column
Python
9 votes225 views1 answer
Override python print for class
Python
1 vote217 views1 answer
Loop error in python
Python
1 vote217 views1 answer
Split a path into all subpaths
Python
0 vote225 views1 answer
How to write stuff in python
Python
0 vote210 views1 answer
Python milliseconds to datetime
Python
16 votes235 views4 answers
Random.choice
Python