[Apache TVM Discuss] [Questions] Nn.dense attribute question

[Chris Gyurgyik via Apache TVM Discuss <no...@discuss.tvm.ai>]

https://tvm.apache.org/docs/api/python/relay/nn.html#tvm.relay.nn.dense

Can someone explain to me what `units` is supposed to represent here? I'm implementing my own `nn.dense` function to be lowered from Relay IR, and I'm unsure what this attribute is supposed to be used for. For context, my background in ML is almost none, so I had trouble finding further information since other dense functions I've seen don't have a `units` attribute.

So far I've seen it used in the VGG_Net example:
```
 let %x139 = Tensor[(1, 4096)] ...;
 let %fc8_weight = Tensor[(10, 4096)] ...;
 let %x140: Tensor[(1, 10), int32] = nn.dense(%x139, %fc8_weight, units=10);
```

Also, if the default is `None`, then what occurs in that case? Maybe that is simple to answer after the explanation :)





---
[Visit Topic](https://discuss.tvm.apache.org/t/nn-dense-attribute-question/8509/1) to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/4c888ed4053700cede7809cb846b9ad8881ad6799fb55765f6506c3401b243cc).