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

14 votes181 views2 answers
Python get username
Python
0 vote183 views1 answer
Datetime day deutsch python
Python
8 votes178 views1 answer
Use beautifulsoup
Python
1 vote170 views2 answers
Converting an array to linked List
Python
0 vote172 views1 answer
How to define a functio in python
Python
0 vote167 views1 answer
Inherit init method
Python
0 vote186 views3 answers
Django add user to group
Python
10 votes184 views6 answers
Variable in python
Python
5 votes167 views3 answers
Click a button using selenium python
Python
2 votes177 views1 answer
Variable in regex python
Python