Error on page 380

[baifanhorst]

In the middle of page 380, it says 'Then we create a joint dataset of t_x and t_y with the custom Dataset class as follows...'. However, the codes below this line is

from torch.utils.data import TensorDataset
joint_dataset = TensorDataset(t_x, t_y)

These cdoes create a dataset with PyTorch's TensorDataset, instead of the manually defined JointDataset class.

The correct code should be:
joint_dataset = JointDataset(t_X, t_y)