Skip to content

Commit f32f081

Browse files
committed
Refer to @DrJonnyT's gist
1 parent 14c2bed commit f32f081

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

docs/advanced_usage.md

+6
Original file line numberDiff line numberDiff line change
@@ -116,3 +116,9 @@ xc_grid_merged, yc_grid_merged = nps.merge_contexts(
116116
B.concat(xt, xt, axis=0)
117117
)
118118
```
119+
120+
## Equivalence of PyTorch and TensorFlow Architectures
121+
122+
Jonny Taylor has a created a very helpful [Gist](https://gist.github.com/DrJonnyT/c946044591fb4ce922b0f5e7fd0f047a)
123+
which can be used to verify the equivalence of PyTorch and TensorFlow versions of architectures.
124+

docs/basic_usage.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -189,4 +189,4 @@ print(dist.logpdf(B.randn(tf.float32, 16, 3, 15)))
189189
print(dist.sample())
190190
print(dist.kl(dist))
191191
print(dist.entropy())
192-
```
192+
```

0 commit comments

Comments
 (0)