API tools faq
paste
Advertisement
Guest User

Untitled

a guest
Jan 28th, 2020
103
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 39.27 KB | None | 0 0
raw download clone embed print report
  1. graph(%x.1 : Tensor,
  2. %u.1 : Tensor,
  3. %k.1 : Tensor,
  4. %dim.1 : int):
  5. %82 : bool = prim::Constant[value=0]()
  6. %81 : int = prim::Constant[value=0]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:202:27
  7. %77 : float = prim::Constant[value=1e-05]()
  8. %76 : float = prim::Constant[value=1e-08]()
  9. %75 : str = prim::Constant[value="Exception"]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:86:8
  10. %74 : int = prim::Constant[value=-1]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:16
  11. %73 : float = prim::Constant[value=0.5]() # /home/ferres/dev/geoopt/geoopt/utils.py:74:33
  12. %72 : int = prim::Constant[value=5]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:31
  13. %71 : int = prim::Constant[value=3]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:59:18
  14. %69 : int = prim::Constant[value=15]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:18
  15. %68 : int = prim::Constant[value=17]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:14
  16. %67 : int = prim::Constant[value=315]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:19
  17. %66 : int = prim::Constant[value=7]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:39
  18. %65 : int = prim::Constant[value=62]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:14
  19. %64 : int = prim::Constant[value=2835]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:19
  20. %63 : int = prim::Constant[value=4]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:31
  21. %62 : int = prim::Constant[value=9]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:40
  22. %61 : int = prim::Constant[value=1382]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:14
  23. %60 : int = prim::Constant[value=155925]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:21
  24. %59 : int = prim::Constant[value=11]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:44
  25. %58 : int = prim::Constant[value=-15]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:31:19
  26. %48 : int = prim::Constant[value=1]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:352:16
  27. %46 : None = prim::Constant()
  28. %7 : bool = prim::Constant[value=1]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1017:42
  29. %6 : int = prim::Constant[value=2]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1017:31
  30. %10 : float = prim::Constant[value=1e-15]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1017:58
  31. %19 : float = prim::Constant[value=2]() # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1019:31
  32. %8 : int[] = prim::ListConstruct(%dim.1)
  33. %9 : Tensor = aten::norm(%u.1, %6, %8, %7) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1017:13
  34. %u_norm.1 : Tensor = aten::clamp_min(%9, %10) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1017:13
  35. %50 : Tensor = aten::pow(%x.1, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:352:24
  36. %51 : int[] = prim::ListConstruct(%dim.1)
  37. %52 : Tensor = aten::sum(%50, %51, %7, %46) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:352:24
  38. %53 : Tensor = aten::mul(%k.1, %52) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:352:20
  39. %54 : Tensor = aten::add(%53, %48, %48) # <string>:5:9
  40. %55 : Tensor = aten::clamp_min(%54, %10) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:352:16
  41. %56 : Tensor = aten::reciprocal(%55) # <string>:21:9
  42. %lam.1 : Tensor = aten::mul(%56, %6) # <string>:21:9
  43. %20 : Tensor = aten::div(%lam.1, %19) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1019:25
  44. %22 : Tensor = aten::mul(%20, %u_norm.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1019:25
  45. %84 : Tensor = prim::Uninitialized()
  46. %k_sign.1 : Tensor = aten::sign(%k.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:192:13
  47. %86 : Device = prim::device(%k.1)
  48. %87 : int = prim::dtype(%k.1)
  49. %88 : int[] = prim::ListConstruct()
  50. %zero.1 : Tensor = aten::zeros(%88, %87, %46, %86, %46) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:193:11
  51. %k_zero.1 : Tensor = aten::isclose(%k.1, %zero.1, %77, %76, %82) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:194:13
  52. %91 : int = prim::dtype(%k_sign.1)
  53. %92 : Tensor = aten::to(%zero.1, %91, %82, %82) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:196:47
  54. %k_sign.4 : Tensor = aten::masked_fill(%k_sign.1, %k_zero.1, %92) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:196:13
  55. %94 : Tensor = aten::all(%k_zero.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:197:7
  56. %95 : bool = aten::Bool(%94) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:197:7
  57. %96 : bool, %97 : Tensor = prim::If(%95) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:197:4
  58. block0():
  59. %98 : Tensor = prim::Uninitialized()
  60. %99 : bool = aten::eq(%48, %81) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:53:7
  61. %100 : bool, %101 : Tensor = prim::If(%99) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:53:4
  62. block0():
  63. -> (%7, %22)
  64. block1():
  65. -> (%82, %98)
  66. %102 : Tensor = prim::If(%100)
  67. block0():
  68. -> (%101)
  69. block1():
  70. %103 : Tensor = aten::sign(%k.1) # /home/ferres/dev/geoopt/geoopt/utils.py:74:22
  71. %104 : Tensor = aten::add(%103, %73, %48) # /home/ferres/dev/geoopt/geoopt/utils.py:74:22
  72. %105 : Tensor = aten::sign(%104) # /home/ferres/dev/geoopt/geoopt/utils.py:74:11
  73. %k.4 : Tensor = aten::mul(%k.1, %105) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:48:11
  74. %107 : bool = aten::eq(%48, %74) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:7
  75. %108 : bool = prim::If(%107) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:7
  76. block0():
  77. -> (%7)
  78. block1():
  79. %109 : bool = aten::eq(%48, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:22
  80. -> (%109)
  81. %110 : Tensor = prim::If(%108) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:4
  82. block0():
  83. %111 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:59:14
  84. %112 : Tensor = aten::mul(%k.4, %111) # <string>:3:9
  85. %113 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:59:26
  86. %114 : Tensor = aten::mul(%112, %113) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:59:14
  87. %115 : Tensor = aten::add(%22, %114, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  88. %116 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:14
  89. %117 : Tensor = aten::pow(%k.4, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:23
  90. %118 : Tensor = aten::mul(%117, %116) # <string>:3:9
  91. %119 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:32
  92. %120 : Tensor = aten::mul(%118, %119) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:14
  93. %121 : Tensor = aten::add(%115, %120, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  94. %122 : float = aten::div(%68, %67) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:14
  95. %123 : Tensor = aten::pow(%k.4, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:25
  96. %124 : Tensor = aten::mul(%123, %122) # <string>:3:9
  97. %125 : Tensor = aten::pow(%22, %66) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:34
  98. %126 : Tensor = aten::mul(%124, %125) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:14
  99. %127 : Tensor = aten::add(%121, %126, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  100. %128 : float = aten::div(%65, %64) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:14
  101. %129 : Tensor = aten::pow(%k.4, %63) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:26
  102. %130 : Tensor = aten::mul(%129, %128) # <string>:3:9
  103. %131 : Tensor = aten::pow(%22, %62) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:35
  104. %132 : Tensor = aten::mul(%130, %131) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:14
  105. %133 : Tensor = aten::add(%127, %132, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  106. %134 : float = aten::div(%61, %60) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:14
  107. %135 : Tensor = aten::pow(%k.4, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:30
  108. %136 : Tensor = aten::mul(%135, %134) # <string>:3:9
  109. %137 : Tensor = aten::pow(%22, %59) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:39
  110. %138 : Tensor = aten::mul(%136, %137) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:14
  111. %139 : Tensor = aten::add(%133, %138, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  112. -> (%139)
  113. block1():
  114. %140 : bool = aten::eq(%48, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:66:9
  115. %141 : Tensor = prim::If(%140) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:66:9
  116. block0():
  117. %142 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:19
  118. %143 : Tensor = aten::mul(%k.4, %142) # <string>:3:9
  119. %144 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:31
  120. %145 : Tensor = aten::mul(%143, %144) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:19
  121. %146 : Tensor = aten::add(%22, %145, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:15
  122. -> (%146)
  123. block1():
  124. %147 : bool = aten::eq(%48, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:68:9
  125. %148 : Tensor = prim::If(%147) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:68:9
  126. block0():
  127. %149 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:19
  128. %150 : Tensor = aten::mul(%k.4, %149) # <string>:3:9
  129. %151 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:31
  130. %152 : Tensor = aten::mul(%150, %151) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:19
  131. %153 : Tensor = aten::add(%22, %152, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:15
  132. %154 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:40
  133. %155 : Tensor = aten::pow(%k.4, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:49
  134. %156 : Tensor = aten::mul(%155, %154) # <string>:3:9
  135. %157 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:58
  136. %158 : Tensor = aten::mul(%156, %157) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:40
  137. %159 : Tensor = aten::add(%153, %158, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:15
  138. -> (%159)
  139. block1():
  140. %160 : bool = aten::eq(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:70:9
  141. %161 : Tensor = prim::If(%160) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:70:9
  142. block0():
  143. %162 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:73:14
  144. %163 : Tensor = aten::mul(%k.4, %162) # <string>:3:9
  145. %164 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:73:26
  146. %165 : Tensor = aten::mul(%163, %164) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:73:14
  147. %166 : Tensor = aten::add(%22, %165, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:72:12
  148. %167 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:14
  149. %168 : Tensor = aten::pow(%k.4, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:23
  150. %169 : Tensor = aten::mul(%168, %167) # <string>:3:9
  151. %170 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:32
  152. %171 : Tensor = aten::mul(%169, %170) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:14
  153. %172 : Tensor = aten::add(%166, %171, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:72:12
  154. %173 : float = aten::div(%68, %67) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:14
  155. %174 : Tensor = aten::pow(%k.4, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:25
  156. %175 : Tensor = aten::mul(%174, %173) # <string>:3:9
  157. %176 : Tensor = aten::pow(%22, %66) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:34
  158. %177 : Tensor = aten::mul(%175, %176) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:14
  159. %178 : Tensor = aten::add(%172, %177, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:72:12
  160. -> (%178)
  161. block1():
  162. %179 : bool = aten::eq(%48, %63) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:77:9
  163. %180 : Tensor = prim::If(%179) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:77:9
  164. block0():
  165. %181 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:80:14
  166. %182 : Tensor = aten::mul(%k.4, %181) # <string>:3:9
  167. %183 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:80:26
  168. %184 : Tensor = aten::mul(%182, %183) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:80:14
  169. %185 : Tensor = aten::add(%22, %184, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  170. %186 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:14
  171. %187 : Tensor = aten::pow(%k.4, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:23
  172. %188 : Tensor = aten::mul(%187, %186) # <string>:3:9
  173. %189 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:32
  174. %190 : Tensor = aten::mul(%188, %189) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:14
  175. %191 : Tensor = aten::add(%185, %190, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  176. %192 : float = aten::div(%68, %67) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:14
  177. %193 : Tensor = aten::pow(%k.4, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:25
  178. %194 : Tensor = aten::mul(%193, %192) # <string>:3:9
  179. %195 : Tensor = aten::pow(%22, %66) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:34
  180. %196 : Tensor = aten::mul(%194, %195) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:14
  181. %197 : Tensor = aten::add(%191, %196, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  182. %198 : float = aten::div(%65, %64) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:14
  183. %199 : Tensor = aten::pow(%k.4, %63) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:26
  184. %200 : Tensor = aten::mul(%199, %198) # <string>:3:9
  185. %201 : Tensor = aten::pow(%22, %62) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:35
  186. %202 : Tensor = aten::mul(%200, %201) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:14
  187. %203 : Tensor = aten::add(%197, %202, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  188. -> (%203)
  189. block1():
  190. = prim::RaiseException(%75) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:86:8
  191. -> (%98)
  192. -> (%180)
  193. -> (%161)
  194. -> (%148)
  195. -> (%141)
  196. -> (%110)
  197. -> (%7, %102)
  198. block1():
  199. -> (%82, %84)
  200. %204 : Tensor = prim::If(%96)
  201. block0():
  202. -> (%97)
  203. block1():
  204. %205 : Tensor = aten::abs(%k.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:199:13
  205. %206 : Tensor = aten::clamp_min(%205, %10) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:199:13
  206. %k_sqrt.1 : Tensor = aten::sqrt(%206) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:199:13
  207. %scaled_x.1 : Tensor = aten::mul(%22, %k_sqrt.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:200:15
  208. %209 : Tensor = aten::lt(%k_sign.4, %81) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:202:17
  209. %210 : Tensor = aten::all(%209) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:202:7
  210. %211 : bool = aten::Bool(%210) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:202:7
  211. %212 : Tensor = prim::If(%211) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:202:4
  212. block0():
  213. %213 : Tensor = aten::reciprocal(%k_sqrt.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:203:15
  214. %214 : Tensor = aten::clamp(%scaled_x.1, %58, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:31:11
  215. %215 : Tensor = aten::tanh(%214) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:31:11
  216. %216 : Tensor = aten::mul(%213, %215) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:203:15
  217. -> (%216)
  218. block1():
  219. %217 : Tensor = aten::gt(%k_sign.4, %81) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:204:19
  220. %218 : Tensor = aten::all(%217) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:204:9
  221. %219 : bool = aten::Bool(%218) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:204:9
  222. %220 : Tensor = prim::If(%219) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:204:9
  223. block0():
  224. %221 : Tensor = aten::reciprocal(%k_sqrt.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:205:15
  225. %222 : Tensor = aten::tan(%scaled_x.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:205:37
  226. %223 : Tensor = aten::mul(%221, %222) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:205:15
  227. -> (%223)
  228. block1():
  229. %224 : Tensor = aten::gt(%k_sign.4, %81) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:208:24
  230. %225 : Tensor = aten::tan(%scaled_x.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:208:38
  231. %226 : Tensor = aten::clamp(%scaled_x.1, %58, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:31:11
  232. %227 : Tensor = aten::tanh(%226) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:31:11
  233. %228 : Tensor = aten::where(%224, %225, %227) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:208:12
  234. %229 : Tensor = aten::reciprocal(%k_sqrt.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:209:14
  235. %tan_k_nonzero.1 : Tensor = aten::mul(%228, %229) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:208:12
  236. %231 : Tensor = prim::Uninitialized()
  237. %232 : bool = aten::eq(%48, %81) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:53:7
  238. %233 : bool, %234 : Tensor = prim::If(%232) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:53:4
  239. block0():
  240. -> (%7, %22)
  241. block1():
  242. -> (%82, %231)
  243. %235 : Tensor = prim::If(%233)
  244. block0():
  245. -> (%234)
  246. block1():
  247. %236 : Tensor = aten::sign(%k.1) # /home/ferres/dev/geoopt/geoopt/utils.py:74:22
  248. %237 : Tensor = aten::add(%236, %73, %48) # /home/ferres/dev/geoopt/geoopt/utils.py:74:22
  249. %238 : Tensor = aten::sign(%237) # /home/ferres/dev/geoopt/geoopt/utils.py:74:11
  250. %k.3 : Tensor = aten::mul(%k.1, %238) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:48:11
  251. %240 : bool = aten::eq(%48, %74) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:7
  252. %241 : bool = prim::If(%240) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:7
  253. block0():
  254. -> (%7)
  255. block1():
  256. %242 : bool = aten::eq(%48, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:22
  257. -> (%242)
  258. %243 : Tensor = prim::If(%241) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:56:4
  259. block0():
  260. %244 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:59:14
  261. %245 : Tensor = aten::mul(%k.3, %244) # <string>:3:9
  262. %246 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:59:26
  263. %247 : Tensor = aten::mul(%245, %246) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:59:14
  264. %248 : Tensor = aten::add(%22, %247, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  265. %249 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:14
  266. %250 : Tensor = aten::pow(%k.3, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:23
  267. %251 : Tensor = aten::mul(%250, %249) # <string>:3:9
  268. %252 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:32
  269. %253 : Tensor = aten::mul(%251, %252) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:60:14
  270. %254 : Tensor = aten::add(%248, %253, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  271. %255 : float = aten::div(%68, %67) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:14
  272. %256 : Tensor = aten::pow(%k.3, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:25
  273. %257 : Tensor = aten::mul(%256, %255) # <string>:3:9
  274. %258 : Tensor = aten::pow(%22, %66) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:34
  275. %259 : Tensor = aten::mul(%257, %258) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:61:14
  276. %260 : Tensor = aten::add(%254, %259, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  277. %261 : float = aten::div(%65, %64) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:14
  278. %262 : Tensor = aten::pow(%k.3, %63) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:26
  279. %263 : Tensor = aten::mul(%262, %261) # <string>:3:9
  280. %264 : Tensor = aten::pow(%22, %62) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:35
  281. %265 : Tensor = aten::mul(%263, %264) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:62:14
  282. %266 : Tensor = aten::add(%260, %265, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  283. %267 : float = aten::div(%61, %60) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:14
  284. %268 : Tensor = aten::pow(%k.3, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:30
  285. %269 : Tensor = aten::mul(%268, %267) # <string>:3:9
  286. %270 : Tensor = aten::pow(%22, %59) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:39
  287. %271 : Tensor = aten::mul(%269, %270) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:63:14
  288. %272 : Tensor = aten::add(%266, %271, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:58:12
  289. -> (%272)
  290. block1():
  291. %273 : bool = aten::eq(%48, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:66:9
  292. %274 : Tensor = prim::If(%273) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:66:9
  293. block0():
  294. %275 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:19
  295. %276 : Tensor = aten::mul(%k.3, %275) # <string>:3:9
  296. %277 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:31
  297. %278 : Tensor = aten::mul(%276, %277) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:19
  298. %279 : Tensor = aten::add(%22, %278, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:67:15
  299. -> (%279)
  300. block1():
  301. %280 : bool = aten::eq(%48, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:68:9
  302. %281 : Tensor = prim::If(%280) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:68:9
  303. block0():
  304. %282 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:19
  305. %283 : Tensor = aten::mul(%k.3, %282) # <string>:3:9
  306. %284 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:31
  307. %285 : Tensor = aten::mul(%283, %284) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:19
  308. %286 : Tensor = aten::add(%22, %285, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:15
  309. %287 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:40
  310. %288 : Tensor = aten::pow(%k.3, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:49
  311. %289 : Tensor = aten::mul(%288, %287) # <string>:3:9
  312. %290 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:58
  313. %291 : Tensor = aten::mul(%289, %290) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:40
  314. %292 : Tensor = aten::add(%286, %291, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:69:15
  315. -> (%292)
  316. block1():
  317. %293 : bool = aten::eq(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:70:9
  318. %294 : Tensor = prim::If(%293) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:70:9
  319. block0():
  320. %295 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:73:14
  321. %296 : Tensor = aten::mul(%k.3, %295) # <string>:3:9
  322. %297 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:73:26
  323. %298 : Tensor = aten::mul(%296, %297) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:73:14
  324. %299 : Tensor = aten::add(%22, %298, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:72:12
  325. %300 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:14
  326. %301 : Tensor = aten::pow(%k.3, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:23
  327. %302 : Tensor = aten::mul(%301, %300) # <string>:3:9
  328. %303 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:32
  329. %304 : Tensor = aten::mul(%302, %303) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:74:14
  330. %305 : Tensor = aten::add(%299, %304, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:72:12
  331. %306 : float = aten::div(%68, %67) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:14
  332. %307 : Tensor = aten::pow(%k.3, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:25
  333. %308 : Tensor = aten::mul(%307, %306) # <string>:3:9
  334. %309 : Tensor = aten::pow(%22, %66) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:34
  335. %310 : Tensor = aten::mul(%308, %309) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:75:14
  336. %311 : Tensor = aten::add(%305, %310, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:72:12
  337. -> (%311)
  338. block1():
  339. %312 : bool = aten::eq(%48, %63) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:77:9
  340. %313 : Tensor = prim::If(%312) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:77:9
  341. block0():
  342. %314 : float = aten::div(%48, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:80:14
  343. %315 : Tensor = aten::mul(%k.3, %314) # <string>:3:9
  344. %316 : Tensor = aten::pow(%22, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:80:26
  345. %317 : Tensor = aten::mul(%315, %316) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:80:14
  346. %318 : Tensor = aten::add(%22, %317, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  347. %319 : float = aten::div(%6, %69) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:14
  348. %320 : Tensor = aten::pow(%k.3, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:23
  349. %321 : Tensor = aten::mul(%320, %319) # <string>:3:9
  350. %322 : Tensor = aten::pow(%22, %72) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:32
  351. %323 : Tensor = aten::mul(%321, %322) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:81:14
  352. %324 : Tensor = aten::add(%318, %323, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  353. %325 : float = aten::div(%68, %67) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:14
  354. %326 : Tensor = aten::pow(%k.3, %71) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:25
  355. %327 : Tensor = aten::mul(%326, %325) # <string>:3:9
  356. %328 : Tensor = aten::pow(%22, %66) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:34
  357. %329 : Tensor = aten::mul(%327, %328) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:82:14
  358. %330 : Tensor = aten::add(%324, %329, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  359. %331 : float = aten::div(%65, %64) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:14
  360. %332 : Tensor = aten::pow(%k.3, %63) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:26
  361. %333 : Tensor = aten::mul(%332, %331) # <string>:3:9
  362. %334 : Tensor = aten::pow(%22, %62) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:35
  363. %335 : Tensor = aten::mul(%333, %334) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:83:14
  364. %336 : Tensor = aten::add(%330, %335, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:79:12
  365. -> (%336)
  366. block1():
  367. = prim::RaiseException(%75) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:86:8
  368. -> (%231)
  369. -> (%313)
  370. -> (%294)
  371. -> (%281)
  372. -> (%274)
  373. -> (%243)
  374. %337 : Tensor = aten::where(%k_zero.1, %235, %tan_k_nonzero.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:211:15
  375. -> (%337)
  376. -> (%220)
  377. -> (%212)
  378. %28 : Tensor = aten::div(%u.1, %u_norm.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1019:52
  379. %second_term.1 : Tensor = aten::mul(%204, %28) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:1019:18
  380. %343 : Tensor = aten::pow(%x.1, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:507:9
  381. %344 : int[] = prim::ListConstruct(%dim.1)
  382. %x2.1 : Tensor = aten::sum(%343, %344, %7, %46) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:507:9
  383. %346 : Tensor = aten::pow(%second_term.1, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:508:9
  384. %347 : int[] = prim::ListConstruct(%dim.1)
  385. %y2.1 : Tensor = aten::sum(%346, %347, %7, %46) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:508:9
  386. %349 : Tensor = aten::mul(%x.1, %second_term.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:509:10
  387. %350 : int[] = prim::ListConstruct(%dim.1)
  388. %xy.1 : Tensor = aten::sum(%349, %350, %7, %46) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:509:10
  389. %352 : Tensor = aten::mul(%k.1, %6) # <string>:3:9
  390. %353 : Tensor = aten::mul(%352, %xy.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:510:15
  391. %354 : Tensor = aten::neg(%353) # <string>:19:9
  392. %355 : Tensor = aten::add(%354, %48, %48) # <string>:19:9
  393. %356 : Tensor = aten::mul(%k.1, %y2.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:510:28
  394. %357 : Tensor = aten::sub(%355, %356, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:510:11
  395. %358 : Tensor = aten::mul(%357, %x.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:510:11
  396. %359 : Tensor = aten::mul(%k.1, %x2.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:510:47
  397. %360 : Tensor = aten::add(%359, %48, %48) # <string>:5:9
  398. %361 : Tensor = aten::mul(%360, %second_term.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:510:43
  399. %num.1 : Tensor = aten::add(%358, %361, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:510:11
  400. %363 : Tensor = aten::mul(%k.1, %6) # <string>:3:9
  401. %364 : Tensor = aten::mul(%363, %xy.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:511:16
  402. %365 : Tensor = aten::neg(%364) # <string>:19:9
  403. %366 : Tensor = aten::add(%365, %48, %48) # <string>:19:9
  404. %367 : Tensor = aten::pow(%k.1, %6) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:511:29
  405. %368 : Tensor = aten::mul(%367, %x2.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:511:29
  406. %369 : Tensor = aten::mul(%368, %y2.1) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:511:29
  407. %denom.1 : Tensor = aten::add(%366, %369, %48) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:511:12
  408. %371 : Tensor = aten::clamp_min(%denom.1, %10) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:524:17
  409. %y.1 : Tensor = aten::div(%num.1, %371) # /home/ferres/dev/geoopt/geoopt/manifolds/stereographic/math.py:524:11
  410. return (%y.1)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
Public Pastes
Advertisement
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!