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 vote154 views2 answers
Add item to tuple
Python
0 vote167 views1 answer
Filter list of tuples python
Python
0 vote167 views1 answer
Pyton count senteses in a text file
Python
0 vote175 views1 answer
Division of 2 numbers in python
Python
4 votes180 views2 answers
How to concat on the basis of particular columns in pandas
Python
0 vote189 views1 answer
Python elasticsearch docker from within other container
Python
0 vote185 views1 answer
Bolumden kalan python
Python
1 vote206 views1 answer
Calculate integral python
Python
122 votes175 views6 answers
Rename columns in python
Python
0 vote179 views2 answers
Set operators
Python