Re:Advent of Code 2021 in pure TensorFlow – day 1
by albertzeyer on Saturday December 11, 2021 @06:50PM
(#29525463)
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.
IMO it's a huge advantage to be able to write torch model code that looks like how you'd write the same program in pure python, and still have them be serializable.