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).
