Task: Add a New Dimension to a PyTorch Tensor

Task: Add a New Dimension to a PyTorch Tensor: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that demonstrates how to add a new dimension (axis) to an existing PyTorch tensor, equivalent to NumPy's np.newaxis functionality.