Skip to content

Commit d42a6ca

Browse files
authored
Tensorflow 2.x backend: multiple outputs extension of DeepONet (#1533)
1 parent 34a4351 commit d42a6ca

File tree

2 files changed

+364
-57
lines changed

2 files changed

+364
-57
lines changed

deepxde/nn/tensorflow/__init__.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,13 @@
11
"""Package for tensorflow NN modules."""
22

3-
__all__ = ["DeepONet", "DeepONetCartesianProd", "FNN", "NN", "PFNN", "PODDeepONet"]
3+
__all__ = [
4+
"DeepONet",
5+
"DeepONetCartesianProd",
6+
"FNN",
7+
"NN",
8+
"PFNN",
9+
"PODDeepONet",
10+
]
411

512
from .deeponet import DeepONet, DeepONetCartesianProd, PODDeepONet
613
from .fnn import FNN, PFNN

0 commit comments

Comments
 (0)