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 vote246 views2 answers
Style django forms with crisp
Python Django
0 vote233 views1 answer
Pandas dataframe froms string
Python
0 vote209 views9 answers
Python code checker and corrector
Python
1 vote272 views7 answers
LightGBM
Python
14 votes234 views3 answers
Getting ip address based on phone number python
Python
4 votes287 views1 answer
How to make a text input box python pygame
Python
0 vote253 views1 answer
Bar plot bokeh
Python
0 vote235 views1 answer
Anvil get last row of data table
Python
3 votes238 views1 answer
Read .dat python
Python
78 votes263 views6 answers
Save dataframe as csv
Python