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

79 votes295 views4 answers
Write a file python
Python
1 vote320 views1 answer
Add background image in django uploaded file
Python Django
0 vote268 views1 answer
OfficeApi
Python
0 vote286 views1 answer
Python ordering items in a list
Python
12 votes311 views5 answers
Pause program python
Python
5 votes272 views1 answer
Get request body flask
Python
7 votes288 views4 answers
Extract year from datetime pandas
Python
1 vote316 views1 answer
Hex to binary python3
Python
0 vote279 views1 answer
How to send more than one variables to python using xlwings
Python
1 vote288 views1 answer
How to provide default value when assign i ngvariables python
Python