search

MokkeMeguru / TFGENZOO

Library about construction helper for Generative models e.g. Flow-based Model with Tensorflow 2.x.
https://mokkemeguru.github.io/TFGENZOO/
12 stars 2 forks source link

Why shift in affine coupling has one more dimension then scale #69

Closed gitlabspy closed 4 years ago

gitlabspy commented 4 years ago

https://github.com/MokkeMeguru/TFGENZOO/blob/001cbb2e3dd01a734b1c9d80d971ea82f7befca2/TFGENZOO/flows/affine_coupling.py#L83 [0:2] while scale `[1:2]'. Shouldn't be same size for both of them like '[0:1]' and '[1:2]'?

gitlabspy commented 4 years ago

I was totally blind.. So sorry...

MokkeMeguru commented 4 years ago

No problem, I'm happy that your doubts have been resolved :D