Why Unsqueezed() X?

[shmuhammadd]

Hi, why do we need to unsequeeze() this In chapter2?

X = torch.arange(start, end, step)#.unsqueeze(dim=1)

[mrdbourke]

Hi @shmuhammad2004,

unsqueeze() removes all single dimensions from a tensor.

Have you tried running the code with and without unsqueeze()? What happens to the inputs and outputs?