Hacker News new | ask | show | jobs
by albertzeyer 1649 days ago
Well, almost exactly the same would work in TensorFlow as well.

But the author choose to make the TF logic more explicitly visible, and use abstractions like tf.Module (like torch.nn.Module), the dataset, etc. I'm not sure this really makes so much sense. I guess this is more for just playing around.