Fix the shape of the outcome in the example code.

[ombre5733]

The toy language uses element-wise multiplication. Transposing and multiplying two tensors with shape <2, 3> gives a tensor with shape <3, 2>.