昇腾社区首页
中文
注册

AscendLinearTransform

API定义

explicit AscendLinearTransform(int dimIn = 0, int dimOut = 0, bool haveBias = false, AscendVectorTransformConfig config = AscendVectorTransformConfig())

功能描述

AscendLinearTransform的构造函数,配置对应参数。

输入

int dimIn:VectorTransform的输入维度。

int dimOut:VectorTransform的输出维度。

bool haveBias:是否添加Bias偏置。

AscendVectorTransformConfig:VectorTransform的配置参数。

输出

返回值

约束说明

dinIn∈{32, 64, 128, 256, 384, 512, 1024}

dinOut∈{32, 64, 128, 256, 384, 512, 1024}

考虑到前向兼容问题,“dimIn”“dimOut”与Faiss保持一致,保留默认值 0,在程序运行阶段不支持维度为 0 的情况。