Error in Week4 / NST

[fotisk07]

When computing the triplet_loss your function outputs 350 while it should output 528

[nishant-sethi]

It should output 528 if we are computing square on axis=-1, which is not the requirement of this notebook. So, I feel this is the only reason why the output varies with yours.