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

3 votes92 views1 answer
SQLalchemy delete by id
Python
3 votes122 views1 answer
Flask upload
Python Flask
0 vote101 views1 answer
Spread dictionary python
Python
3 votes102 views1 answer
Drop all characters after a character in python
Python
0 vote263 views1 answer
Python local date time
Python
1 vote96 views1 answer
Fatal error detected failed to execute script
Python
10 votes101 views4 answers
Create empty numpy array
Python
0 vote97 views1 answer
How to find 2 similar words in atext python
Python
1 vote88 views2 answers
Numpy random for string
Python
66 votes63 views2 answers
'pip' is not recognized as an internal or external command, operable program or batch file.
Python