模型由EmbeddingBag层和线性层组成。nn.EmbeddingBag计算embedding包的均值。 ... EmbeddingBag无需padding,因为文本长度在offsets已保存。
確定! 回上一頁