Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ainode处理带有 batchsize三维张量 #14112

Open
YWzzq opened this issue Nov 17, 2024 · 2 comments
Open

ainode处理带有 batchsize三维张量 #14112

YWzzq opened this issue Nov 17, 2024 · 2 comments

Comments

@YWzzq
Copy link

YWzzq commented Nov 17, 2024

ainode部署模型需要知道模型输入大小,二维的,但我的模型会有一层batchsize,怎么办

@CritasWang
Copy link
Collaborator

没太理解,可以具体一些吗

@ycycse
Copy link
Member

ycycse commented Nov 27, 2024

AINode 在推理时,会通过 unsqueeze 方法,将从数据库中查到的二维数据(L * C)转为三维数据(1 * L * C)。所以你不用担心 batchsize 这个额外的维度问题,会被统一处理,只要关注前面接受的时序数据的行列数就好

@ycycse ycycse changed the title ainode部署 ainode处理带有 batchsize三维张量 Nov 27, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants