Иначе во всех примерах (в колабе по крайней мере), вылетала ошибка: ValueError: Expected 2D Tensor, got 1D.