zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-18 10:26:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Docs updates (#198)

Browse Source 
Reorganize NN docs + a few other tidbits.
This commit is contained in:
Awni Hannun
2023-12-17 13:20:55 -08:00
committed by GitHub
parent 90d04072b7
commit ee0c2835c5
9 changed files with 160 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
python/mlx/nn/layers/linear.py
View File

@@ -7,12 +7,21 @@ from mlx.nn.layers.base import Module
class Linear(Module):
"""Applies an affine transformation to the input.
r"""Applies an affine transformation to the input.
Concretely:
.. math::
y = W^\top x + b
where :math:`W` has shape ``[output_dims, input_dims]``.
Args:
input_dims (int): The dimensionality of the input features
output_dims (int): The dimensionality of the output features
bias (bool): If set to False then the layer will not use a bias
bias (bool, optional): If set to ``False`` then the layer will
not use a bias. Default ``True``.
"""
def __init__(self, input_dims: int, output_dims: int, bias: bool = True):