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

11 votes217 views6 answers
Np.random.randint
Python
0 vote343 views1 answer
How to install flask module in vscode
Python Flask
5 votes276 views2 answers
Get_object_or_404 django
Python Django
11 votes225 views3 answers
How to get the length of a string in python
Python
18 votes199 views1 answer
Pandas read txt
Python
0 vote240 views1 answer
Onlinecourses.osppro.com
Python
5 votes231 views1 answer
Python3 add dictionary to dictionary
Python
0 vote203 views1 answer
Displaying cv2.imshow on specific window position
Python
0 vote239 views1 answer
Windows error message python
Python
0 vote231 views1 answer
Python array linspace
Python