Polymorphism example code

[anj1]

In polymorphism.jl, should the following code:

for i in 1:batch_size
  y_one_hot[y, i] = 1.
end

instead be:

for i in 1:batch_size
  y_one_hot[y[i], i] = 1.
end

? I'm trying to see if I understand the example correctly.