Emulating logic gates with binary Neural network?

Greetings, I’m attempting to learn Pytorch by making a Neural network which takes
some binary input ( In the form of a torch.bool tensor )
What layers accept a torch.bool And/Or are the best fit for logic gates emulation/estimation?

Any suggestions?