3 ms·

Pytorch makes it even more fun.