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

5 votes131 views5 answers
How to print upto 2 decimal places in python
Python
3 votes218 views1 answer
Python distance of coordinates
Python
1 vote242 views1 answer
Curso python
Python
2 votes187 views2 answers
Make a list in python 3
Python
0 vote183 views1 answer
How to make a nan value in a list
Python
2 votes175 views1 answer
None address in python
Python
5 votes183 views1 answer
Python flask query params
Python
2 votes184 views1 answer
Not scientific notation python
Python
0 vote172 views1 answer
Downolad fileby python requests
Python
0 vote170 views1 answer
Skip error python
Python