deepmd.tf2.env#

TensorFlow eager backend environment.

Attributes#

Functions#

stop_gradient(→ Any)

Stop gradients on TensorFlow-backed Array objects.

Module Contents#

deepmd.tf2.env.Array[source]#
deepmd.tf2.env.stop_gradient(value: Any) → Any[source]#

Stop gradients on TensorFlow-backed Array objects.