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

6 votes139 views5 answers
How to count unique values in a column dataframe in python
Python
0 vote131 views1 answer
How to tell if member is a bot discord.py
Python
1 vote178 views1 answer
Write a table to CSV file python
Python
1 vote147 views1 answer
Pandas timedelta to seconds
Python
-1 vote134 views1 answer
Return an RDD with the keys of each tuple.
Python
0 vote143 views1 answer
Python eliptic curve matplotlib
Python
0 vote157 views1 answer
Python Textfeld lköschen
Python
1 vote138 views1 answer
How to run different python version
Python
27 votes141 views6 answers
Create a empty dataframe
Python
3 votes165 views1 answer
Python bounding box on image
Python