Shouldn't [0,2,1] = 5 on page 17 and not 2? Or do i misunderstand how pytorch handle tensors?