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

2 votes257 views1 answer
How to launch jupyter notebook from cmd
Python
2 votes264 views3 answers
Find difference between two pandas dataframes
Python
17 votes262 views3 answers
Python degrees to radians
Python
0 vote279 views1 answer
Pandas order dataframe by index of other dataframe
Python
0 vote274 views1 answer
Merge multile dict
Python
0 vote247 views1 answer
Start of the american labor movement
Python
2 votes277 views1 answer
Clock in python
Python
0 vote232 views1 answer
Check string in a list for substrings and return index
Python
0 vote255 views1 answer
Renpy quickstart
Python
1 vote257 views1 answer
Enumerate string pythonm
Python