Incorrect solution for Part 1 - Tensors in PyTorch(Solutions).ipynb (v2)

[niazangels]

The notebook states that the answer is:

h = activation(torch.mm(features, W1)) + B1
output = activation(torch.mm(h, W2)) + B2

The output evaluates to tensor([[ 0.4813]]) which is also mentioned as the correct solution in the markdown in the below cell.

The correct answer, however, should be:

h = activation(torch.mm(features, W1) + B1)
output = activation(torch.mm(h1, W2) + B2)

which evaluates to tensor([[0.3171]]).

The video solution shows the correct code as the latter, but the cell is never run in the video segment. The expected answer as seen in the notebook from the video is tensor([[ 0.4813]]).