Skip to content

Commit

Permalink
refactor api.md (d2l-ai#1894)
Browse files Browse the repository at this point in the history
* re-org d2l.md

* update api.md
  • Loading branch information
mli authored Aug 24, 2021
1 parent fb5637e commit 63cafb0
Show file tree
Hide file tree
Showing 4 changed files with 1,639 additions and 1,853 deletions.
78 changes: 67 additions & 11 deletions chapter_appendix-tools-for-deep-learning/d2l.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,7 @@ The implementations of the following members of the `d2l` package and sections w

```eval_rst
.. automodule:: d2l.mxnet
:members:
:imported-members:
.. currentmodule:: d2l.mxnet
```

Expand All @@ -20,23 +18,81 @@ The implementations of the following members of the `d2l` package and sections w

```eval_rst
.. automodule:: d2l.torch
:members:
:imported-members:
.. currentmodule:: d2l.torch
```

:begin_tab:`tensorflow`

```eval_rst
.. currentmodule:: d2l.torch
```

:end_tab:

## Models

:begin_tab:`tensorflow`
```eval_rst
```eval_rst
.. autoclass:: Module
:members:
.. automodule:: d2l.tensorflow
.. autoclass:: LinearRegressionScratch
:members:
.. autoclass:: LinearRegression
:members:
.. autoclass:: Classification
:members:
:imported-members:
```

:end_tab:
## Data

```eval_rst
.. autoclass:: DataModule
:members:
.. autoclass:: SyntheticRegressionData
:members:
.. autoclass:: FashionMNIST
:members:
```

## Trainer

```eval_rst
.. autoclass:: Trainer
:members:
.. autoclass:: SGD
:members:
```

## Utilities

```eval_rst
.. autofunction:: add_to_class
.. autofunction:: cpu
.. autofunction:: gpu
.. autofunction:: num_gpus
.. autoclass:: ProgressBoard
:members:
.. autoclass:: HyperParameters
:members:
```
Loading

0 comments on commit 63cafb0

Please sign in to comment.