3 ms·

thanks for adding this context! One caveat is that the PyTorch strides used in the post are defined by elements rather than byte-wise.