mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
52 lines
1,013 B
VimL
52 lines
1,013 B
VimL
"Slightly adjust indentation
|
|
%s/^ / /g
|
|
|
|
" # -> len
|
|
%s/#\(\S*\) /len(\1)/g
|
|
|
|
" for loops
|
|
%s/for\( \)\{-\}\(\S*\)\( \)\{-\}=\( \)\{-\}\(\S*\),\( \)\{-\}\(\S*\)\( \)\{-\}do/for \2 in range(\5, \7+1)/g
|
|
|
|
" Change comments
|
|
%s/--\[\[/"""/g
|
|
%s/]]/"""/g
|
|
%s/--/#/g
|
|
|
|
" Add spacing between commas
|
|
%s/\(\S\),\(\S\)/\1, \2/g
|
|
|
|
%s/local //g
|
|
%s/ then/:/g
|
|
%s/ do/:/g
|
|
%s/end//g
|
|
%s/elseif/elif/g
|
|
%s/else/else:/g
|
|
%s/true/True/g
|
|
%s/false/False/g
|
|
%s/\~=/!=/g
|
|
%s/math\.min/min/g
|
|
%s/math\.max/max/g
|
|
%s/math\.abs/abs/g
|
|
|
|
|
|
%s/__init/__init__/g
|
|
|
|
" Rewrite function declarations
|
|
%s/function \w*:\(\w*\)/ def \1/g
|
|
%s/def \(.*\)$/def \1:/g
|
|
|
|
" class declaration
|
|
%s/\(\w*\), parent = torch\.class.*$/import torch\rfrom torch.legacy import nn\r\rclass \1(nn.Module):/g
|
|
|
|
%s/input\.THNN/self._backend/g
|
|
%s/\(self\.backend\w*$\)/\1\r self._backend.library_state,/g
|
|
%s/def \(\w*\)(/def \1(self, /g
|
|
|
|
%s/__init__(self)/__init__()/g
|
|
|
|
%s/:\(\S\)/.\1/g
|
|
|
|
%s/\.cdata()//g
|
|
%s/THNN\.optionalTensor(\(.*\))/\1/g
|
|
|
|
%s/parent\./super(##, self)./g
|