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

0 vote269 views1 answer
111
Python
22 votes250 views5 answers
Select rows with nan pandas
Python
19 votes242 views1 answer
Python turn of userwarning
Python
3 votes269 views2 answers
Python inline print variable
Python
6 votes270 views3 answers
Pi in python
Python
0 vote246 views2 answers
Remove vowels in a string python
Python
0 vote239 views1 answer
Jouer à Snake
Python
0 vote310 views1 answer
ComplexWarning: Casting complex values to real discards the imaginary part
Python
12 votes289 views5 answers
Pause program python
Python
24 votes276 views6 answers
Display all rows pandas
Python