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 vote408 views1 answer
Streamlit bold
Python
8 votes319 views6 answers
Python get numbers from string
Python
0 vote319 views1 answer
Python use variable name as variable
Python
0 vote398 views1 answer
Flask development mode
Python
0 vote315 views1 answer
How do you see if a data type is an integer python
Python
7 votes348 views3 answers
Mysql connection python
Python
20 votes313 views2 answers
How to import csv in pandas
Python
1 vote301 views1 answer
Codeforces - 570b python
Python
3 votes330 views1 answer
Multiple linear regression in python
Python
0 vote356 views1 answer
Python get factors of a number
Python