We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f403119 commit 2236965Copy full SHA for 2236965
deepxde/nn/tensorflow/deeponet.py
@@ -486,7 +486,7 @@ def __init__(
486
regularization=None,
487
):
488
super().__init__()
489
- self.pod_basis = tf.convert_to_tensor(pod_basis, dtype=tf.float32)
+ self.pod_basis = tf.convert_to_tensor(pod_basis, dtype=config.real(tf))
490
if isinstance(activation, dict):
491
activation_branch = activation["branch"]
492
self.activation_trunk = activations.get(activation["trunk"])
0 commit comments