TensorFlow tf.keras.layers.Embedding
参数
列子:
from tensorflow import keras import numpy as np data = np.array([[0,0,0],[1,1,1]]) emb = keras.layers.Embedding(input_dim=2, output_dim=3, input_length=3) # 只有0,1所以input_dime是2 emb(data)
输入形状
二维张量(batch_size,input_length)
输出形状
三维张量(batch_size,input_length,output_dim)
init
__init__(
input_dim,
output_dim,
embeddings_initializer=uniform,
embeddings_regularizer=None,
activity_regularizer=None,
embeddings_constraint=None,
mask_zero=False,
input_length=None,
**kwargs
)
参考:
