def _make_one_hot(self, word_inds, vec_size):
onehot = numpy.zeros((word_inds.shape + (vec_size,)))
for inds in itertools.product(*[numpy.arange(s) for s in word_inds.shape]):
onehot[inds+(word_inds[inds],)] = 1
return onehot
评论列表
文章目录