# Get first embedding at epoch = 0
embed = h.detach().cpu().numpy()