|
|
|
@ -156,10 +156,10 @@ class Dense(Cell):
|
|
|
|
|
ValueError: If weight_init or bias_init shape is incorrect.
|
|
|
|
|
|
|
|
|
|
Inputs:
|
|
|
|
|
- **input** (Tensor) - Tensor of shape :math:`(N, in_channels)`.
|
|
|
|
|
- **input** (Tensor) - Tensor of shape :math:`(N, in\_channels)`.
|
|
|
|
|
|
|
|
|
|
Outputs:
|
|
|
|
|
Tensor of shape :math:`(N, out_channels)`.
|
|
|
|
|
Tensor of shape :math:`(N, out\_channels)`.
|
|
|
|
|
|
|
|
|
|
Examples:
|
|
|
|
|
>>> net = nn.Dense(3, 4)
|
|
|
|
|