mirror of
https://github.com/zebrajr/pytorch.git
synced 2026-01-15 12:15:51 +00:00
bbcd6cc782
Summary: This adds support for PEP 526 style annotations on assignments in place of `torch.jit.annotate()`, so ```python a = torch.jit.annotate(List[int], []) ``` turns into ```python a : List[int] = [] ``` ](https://our.intern.facebook.com/intern/diff/15706021/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/21390 Pulled By: driazati Differential Revision: D15706021 fbshipit-source-id: 8bf1459f229d5fd0e16e59953b9656e85a2207fb
81 lines
1.6 KiB
Plaintext
81 lines
1.6 KiB
Plaintext
(def
|
|
(ident fn)
|
|
(decl
|
|
(list
|
|
(param
|
|
(ident x)
|
|
(option)
|
|
(option)
|
|
(False))
|
|
(param
|
|
(ident y)
|
|
(option)
|
|
(option)
|
|
(False))
|
|
(param
|
|
(ident z)
|
|
(option)
|
|
(option)
|
|
(False)))
|
|
(option))
|
|
(list
|
|
(assign
|
|
(variable (ident q))
|
|
(None)
|
|
(option))
|
|
(assign
|
|
(variable (ident q))
|
|
(-
|
|
(+
|
|
(variable (ident x))
|
|
(variable (ident y)))
|
|
(apply
|
|
(.
|
|
(variable (ident z))
|
|
(ident sigmoid))
|
|
(list)
|
|
(list)))
|
|
(option))
|
|
(expression statement
|
|
(apply
|
|
(variable (ident print))
|
|
(list (variable (ident q)))
|
|
(list)))
|
|
(assign
|
|
(variable (ident w))
|
|
(unary minus
|
|
(variable (ident z)))
|
|
(option))
|
|
(if
|
|
(and
|
|
(and
|
|
(not (variable (ident x)))
|
|
(not (variable (ident y))))
|
|
(variable (ident z)))
|
|
(list
|
|
(assign
|
|
(variable (ident m))
|
|
(if
|
|
(not (variable (ident z)))
|
|
(variable (ident x))
|
|
(variable (ident y)))
|
|
(option)))
|
|
(list))
|
|
(while
|
|
(and
|
|
(<
|
|
(variable (ident x))
|
|
(variable (ident y)))
|
|
(>
|
|
(variable (ident y))
|
|
(variable (ident z))))
|
|
(list
|
|
(assign
|
|
(variable (ident q))
|
|
(variable (ident x))
|
|
(option))))
|
|
(assert
|
|
(eq (const 1) (const 1))
|
|
(option (string_literal hello)))
|
|
(return (variable (ident x)))))
|