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 vote203 views1 answer
Open file rw python
Python
8 votes157 views3 answers
Django order by
Python
47 votes166 views3 answers
How to change an integer to a string in python permanently
Python
6 votes151 views1 answer
How to get username with userid discord.py
Python
0 vote166 views1 answer
Base conversion python
Python
1 vote158 views1 answer
Python edit string variable
Python
0 vote160 views1 answer
Rest api error
Python
0 vote193 views1 answer
Python imread multiple images
Python
49 votes157 views2 answers
Add new element to python dictionary
Python
15 votes170 views3 answers
How to get the first key of a dictionary in python
Python