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 vote130 views1 answer
Upgrade python to 3.9 i linux
Python
0 vote196 views1 answer
Switch columns and rows python
Python
0 vote131 views1 answer
How to python string up
Python
2 votes138 views2 answers
Create conda env with specific python version
Python
1 vote130 views3 answers
Python solve linear equation system
Python
-1 vote120 views1 answer
Heads or tails python
Python
15 votes117 views5 answers
Check empty dataframe
Python
7 votes134 views2 answers
How to change values of dictionary in python
Python
4 votes147 views1 answer
Django is null
Python Django
0 vote145 views1 answer
Generate random list and find max in list python
Python