I got stuck at this point where I couldn’t find the function that gives outputs for `_, n_feat`

, I can’t determine which note talks about that, any additional note for more clarification?

1 Like

There’s a hint couple of cells above, in the `TripletLoss()`

function implementation:

```
def TripletLoss(labels, out, margin=0.25):
_, embedding_size = out.shape # get embedding size
...
```

Note, that in your case the variable is not named `out`

, but `pred`

. So, to get the dimensionality (n_feat) of the model’s prediction, you just need to use the “shape” property of this variable.

Cheers