deepmd.tf.op._prod_virial_grad

Gradients for prod virial.

Module Contents

Functions

_prod_virial_grad_cc(op, grad, grad_atom)

deepmd.tf.op._prod_virial_grad._prod_virial_grad_cc(op, grad, grad_atom)[source]