如何输出更小的维度

[localoca5]

您好！请问怎么让模型输出更小的维度呢(比如64维)？768维在实际中占用空间太大，而且计算也比较耗时

[zejunwang1]

您好！请问怎么让模型输出更小的维度呢(比如64维)？768维在实际中占用空间太大，而且计算也比较耗时

bert4vec 支持加载的四个模型中，能输出的最小维度是 384，可以使用：

model = Bert4Vec(mode='roformer-sim-small')
model = Bert4Vec(mode='paraphrase-multilingual-minilm')

输出 384 维度的句向量。