ec1c64cf6d6e635fa80e9b4f55bda7ece84b5d81,thinc/wire.py,,with_flatten,#Any#Any#Any#,195

Before Change

        X = layer(layer.ops.flatten(seqs_in, pad=pad))
        return layer.ops.unflatten(X, lengths, pad=pad)

    model = layerize(begin_update, predict=predict)
    model._layers.append(layer)
    model.on_data_hooks.append(_with_flatten_on_data)
    model.name = "flatten"
    return model


def _with_flatten_on_data(model, X, y):
    X = model.ops.flatten(X)

After Change

        X = layer(layer.ops.flatten(seqs_in, pad=pad))
        return layer.ops.unflatten(X, lengths, pad=pad)
    
    return wrap(with_flatten_forward, layer, predict=with_flatten_predict, 
        name=f"with_flatten-{layer.name}", on_data_hooks=[_with_flatten_on_data])


def _with_flatten_on_data(model, X, y):
    X = model.ops.flatten(X)

Instances