We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
import torch import torch.nn as nn import torch.nn.functional as F class Model(nn.Module): def __init__(self): super(Model, self).__init__() def forward(self, x, y): x = torch.tensordot(x, y, dims=([1], [0])) return x model = Model() model.eval() input1 = torch.rand(3, 4) input2 = torch.rand(4, 2) trace = torch.jit.trace(model, (input1, input2)) trace.save('tensordot.pt')
fallback batch axis 233 for operand 0 fallback batch axis 233 for operand 1 fallback batch axis 233 for operand 2 fallback batch axis 233 for operand 3 fallback batch axis 233 for operand 4 ignore pnnx.Expression pnnx_expr_1 param expr=[1] ignore pnnx.Expression pnnx_expr_0 param expr=[0]
The text was updated successfully, but these errors were encountered:
No branches or pull requests
error log | 日志或报错信息 | ログ
model | 模型 | モデル
how to reproduce | 复现步骤 | 再現方法
The text was updated successfully, but these errors were encountered: