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

1 vote178 views5 answers
Python code
Python
0 vote196 views1 answer
Cv2.imwrite path
Python
1 vote145 views1 answer
Python get keys as list
Python
0 vote197 views1 answer
How to populate an image to template in django from views
Python
0 vote234 views1 answer
Openai gym random action
Python
1 vote180 views1 answer
Circular array python
Python
2 votes201 views6 answers
Convert ui to py
Python
2 votes189 views2 answers
Fillna string
Python
0 vote184 views1 answer
How to take first half of list python
Python
0 vote185 views1 answer
Jupyter markdown new line
Python