def gen_onehot_keys(input_size, batch_size, targets):
return np.vstack([one_hot(input_size, [t]) for t in targets])
微信
公众号
扫码关注公众号
微信
公众号