blob: 6a464f862192ea72a0f5acd1e98bb92cdbf60177 (
plain)
1
2
3
4
5
6
7
8
9
10
|
int main()
{
Shape s{1, 2, 3};
Tensor in_t(s);
NNModel model("nnmodel.params");
model.set_in(in_t);
model.doInference();
std::shared_ptr<Tensor> out_t = model.get_out();
return 0;
}
|