tensorflow one_hot

 tensorflow 的one_hot 比较灵活,可以使用depth指定生成one_hot向量的维度,indices长度说明生成one_hot向量的个数,而其中的数字指定在向量的哪个位置中使用on_value(默认1)激活;

axis 指定生成的矩阵的排列方式。

猜你喜欢

转载自www.cnblogs.com/wdmx/p/10140195.html