ValueError: Tensor._shape cannot be assigned, use Tensor.set_shape instead.

[Hhhhhhao]

I'm calculating the mean and std of the training data, and I got the error like this:

here is my code:

[HRamses]

Hi!
It seems to be the same reason described in the previous issue #6. I finally committed the change suggested by foofighter112. I hope it will work now.