pytorch/test/expect/TestJit.test_pretty_printer-if_test.expect at master - saymrwulf/pytorch - Forgejo: Beyond coding. We forge.

saymrwulf/pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00

Zachary DeVito 330990d878 Serialize first-class version of functions (#19723 )

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19723
ghimport-source-id: 7f7ec6200c3b42d19046a3e228a3d82212697f14

Reviewed By: jamesr66a

Differential Revision: D15078533

Pulled By: zdevito

fbshipit-source-id: fe421afab9607ee942f6d200f04bb6335fc0aa97

2019-04-25 15:53:07 -07:00

7 lines

115 B

Text

Raw Permalink Blame History

 def if_test(a: Tensor,
     b: Tensor) -> Tensor:
   if bool(torch.lt(a, b)):
     c = b
   else:
     c = a
   return c