deepmd.tf.nvnmd.descriptor.se_atten
Module Contents
Functions
Get the davg and dstd from the dictionary nvnmd_cfg. | |
| Check the range of switch, let it in range [-2, 14]. |
Replace se_a.py/DescrptSeA/build. | |
| Replace \(r_{ji} \rightarrow r'_{ji}\) |
| Replace se_a.py/DescrptSeA/_filter_lower. |
| Replace se_a.py/_filter. |
Attributes
- deepmd.tf.nvnmd.descriptor.se_atten.build_davg_dstd()[source]
Get the davg and dstd from the dictionary nvnmd_cfg. The davg and dstd have been obtained by training CNN.
- deepmd.tf.nvnmd.descriptor.se_atten.check_switch_range(davg, dstd)[source]
Check the range of switch, let it in range [-2, 14].
- deepmd.tf.nvnmd.descriptor.se_atten.build_op_descriptor()[source]
Replace se_a.py/DescrptSeA/build.
- deepmd.tf.nvnmd.descriptor.se_atten.descrpt2r4(inputs, atype)[source]
Replace \(r_{ji} \rightarrow r'_{ji}\) where \(r_{ji} = (x_{ji}, y_{ji}, z_{ji})\) and \(r'_{ji} = (s_{ji}, \frac{s_{ji} x_{ji}}{r_{ji}}, \frac{s_{ji} y_{ji}}{r_{ji}}, \frac{s_{ji} z_{ji}}{r_{ji}})\).