Use broadcasting instead of tf.tile

hiwonjoon / tf-vqvae

Tensorflow Implementation of the paper [Neural Discrete Representation Learning](https://arxiv.org/abs/1711.00937) (VQ-VAE).

258 stars 41 forks source link

Closed carlthome closed 6 years ago

carlthome commented 6 years ago

Semi-sure this is equivalent but would love some feedback.

hiwonjoon commented 6 years ago

Nice! I tested it and it worked as it is. Thanks for the contribution 👍