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 vote130 views1 answer
Tanimoto coefficient rdkit
Python
4 votes119 views4 answers
Interface graphique sur python
Python
1 vote125 views2 answers
Check if user has manage messages discord.py
Python
10 votes120 views5 answers
Find nan values in a column pandas
Python
0 vote130 views1 answer
Append a line to a text file using the write() function
Python
0 vote109 views1 answer
Check if digit or alphabet
Python
-2 votes106 views5 answers
Wow
Python
77 votes117 views6 answers
Python version
Python
0 vote105 views1 answer
Python apply file line
Python
1 vote107 views1 answer
Python append several values
Python