c638823db4caa6c562423903807b5eac652b2e23,thinc/layers/staticvectors.py,,forward,#,28

Before Change

def forward(model: Model[InT, OutT], ids: InT, is_train: bool) -> Tuple[OutT, Callable]:
    column = model.get_attr("column")
    W = cast(Array2d, model.get_param("W"))
    vector_table = _get_vectors(model.ops, model.get_attr("lang"))
    if ids.ndim >= 2:
        ids = model.ops.as_contig(ids[:, column])
    vectors = vector_table[ids * (ids < vector_table.shape[0])]

After Change

def forward(model: Model[InT, OutT], ids: InT, is_train: bool) -> Tuple[OutT, Callable]:
    if model.has_attr("dropout_rate"):
        dropout = model.get_attr("dropout_rate")
    else:
        dropout = None
    column = model.get_attr("column")
    W = cast(Array2d, model.get_param("W"))

Instances