deepmd.tf.op._soft_min_virial_grad

Gradients for soft min virial.

Functions

_soft_min_virial_grad_cc(...)

Module Contents

deepmd.tf.op._soft_min_virial_grad._soft_min_virial_grad_cc(op: deepmd.tf.env.tf.Operation, grad: deepmd.tf.env.tf.Tensor, grad_atom: deepmd.tf.env.tf.Tensor) → list[deepmd.tf.env.tf.Tensor | None]