SymPy 量子力学模块的 InnerProduct:Bra-Ket 符号内积的实现原理与实战指南
【免费下载链接】sympyA computer algebra system written in pure Python项目地址: https://gitcode.com/GitHub_Trending/sy/sympy
本篇技术指南围绕 SymPy 量子力学模块(sympy.physics.quantum)中的内积组件展开:它通过sympy.physics.quantum.innerproduct模块提供InnerProduct类,用于表示 Bra 与 Ket 之间未求值的符号内积。读者将掌握InnerProduct的构造方式、与 Bra/Ket 乘法的自动识别规则、doit()求值机制、与Dagger的共轭交互,以及如何通过自定义量子态类扩展内积的求值行为,并理解其在处理叠加态、归一化、正交性等量子力学计算中的实际用法。
InnerProduct 在 SymPy 量子力学模块中的定位
InnerProduct隶属于 SymPy 的量子力学子包(参见 Quantum Mechanics 模块索引 中的 "Quantum Functions" 一节),与Commutator、AntiCommutator、Dagger、TensorProduct等量子运算类并列,是 Dirac 符号体系的核心运算组件之一。
在物理语义上,内积 $\langle bra | ket \rangle$ 将两个量子态映射为一个复数(概率幅),其物理意义包括计算跃迁概率、归一化条件、正交性判断等。SymPy 用InnerProduct类承载这一概念:当 Bra 与 Ket 直接相乘(b * k)时,SymPy 会自动识别并构造出内积对象;而当内积暂时无法求出具体数值时,它会以未求值的形式保留在表达式中,等待后续的doit()触发实际计算。
核心类 InnerProduct 的 API 与基础用法
InnerProduct定义在 sympy/physics/quantum/innerproduct.py 中,完整签名如下:
class InnerProduct(Expr): """An unevaluated inner product between a Bra and a Ket.""" kind = NumberKind is_complex = True def __new__(cls, bra, ket): ...构造参数
| 参数 | 类型 | 含义 |
|---|---|---|
bra | BraBase或子类 | 内积左侧的 Bra,即 $\langle bra \mid$ |
ket | KetBase或子类 | 内积右侧的 Ket,即 $\mid ket \rangle$ |
__new__方法(innerproduct.py#L75-L84)对参数类型做了严格校验:若ket不是KetBase子类、bra不是BraBase子类,会分别抛出TypeError。校验通过后委托给Expr.__new__构建实例。注意这里将BraBase/KetBase的导入放在方法内部,是为了避免循环导入问题。
基础属性
对象创建后可通过两个只读属性访问其构成部分(innerproduct.py#L86-L92):
>>> from sympy.physics.quantum import Bra, Ket >>> b = Bra('b') >>> k = Ket('k') >>> ip = b * k >>> ip <b|k> >>> ip.bra <b| >>> ip.ket |k>bra与ket属性分别返回self.args[0]与self.args[1],即内积表达式的两个底层参数。
Bra-Ket 乘法的自动内积识别
在 Dirac 记号中,Bra 与 Ket 相邻相乘 $\langle b | k \rangle$ 即为内积。SymPy 的量子表达式乘法会自动识别这一模式并构造InnerProduct:
>>> b * k <b|k> >>> b * k == InnerProduct(b, k) True值得注意的是一条内积优先规则(innerproduct.py#L56-L63 的类文档示例):在较复杂的表达式中,当内积与外积的识别存在歧义时,内积具有更高优先级:
>>> k * b * k * b <b|k>*|k><b|这里k * b本可以是外积 $\mid k \rangle\langle b \mid$,但系统优先将其识别为内积 $\langle b | k \rangle$,并且因为内积是可交换的复数,它自动移动到整个表达式的最左侧,剩余部分|k><b|才作为外积保留。
设计取舍:为什么 InnerProduct 是 Expr 而非 QExpr
这是理解该模块设计的关键点。innerproduct.py#L16-L21 的注释详细说明了这一决策:InnerProduct不是QExpr子类,而是直接继承普通的Expr,因为内积本质上是普通的交换复数,而非量子对象。开发者曾反复权衡过这一选择,最终因为以下收益而确定:
- 借助
_eval_conjugate使Dagger正常工作; represent环节可以直接使用atoms和subs;- 保持"不存在可交换的 QExpr 子类"这一约束,简化了整个模块的设计。
因此类上设置了kind = NumberKind与is_complex = True,与 QExpr 的is_commutative = False形成鲜明对比。这一性质解释了上节中内积因子在乘积中可任意换位、自动左移的行为。
doit():从符号内积到实际求值
InnerProduct默认是未求值的符号对象,真正计算内积数值/符号结果的方法是doit()(innerproduct.py#L127-L139)。其求值策略分两级:
- 首先尝试调用
self.ket._eval_innerproduct(self.bra, **hints),即询问 Ket 是否知道如何与给定 Bra 计算内积; - 若抛出
NotImplementedError,则回退为对偶方案:计算conjugate(self.bra.dual._eval_innerproduct(self.ket.dual))(利用 $\langle b|k\rangle^* = \langle k|b\rangle$),再取共轭; - 若两级都失败(
NotImplementedError)或结果为None,则原样返回自身,保持未求值状态。
Ket 侧的分发机制
doit()的第一步依赖KetBase._eval_innerproduct(sympy/physics/quantum/state.py#L235-L247),它通过dispatch_method将求值责任分发给形如_eval_innerproduct_<BraClassName>的专用方法。分发器定义在 sympy/physics/quantum/qexpr.py#L398-L410:
def dispatch_method(self, basename, arg, **options): """Dispatch a method to the proper handlers.""" method_name = '%s_%s' % (basename, arg.__class__.__name__) if hasattr(self, method_name): f = getattr(self, method_name) result = f(arg, **options) if result is not None: return result raise NotImplementedError(...)即:若 Ket 类中定义了_eval_innerproduct_FooBra这样的方法,遇到FooBra类型的 Bra 时就会自动调用它,从而实现按 Bra 类型分发的扩展机制。
内置示例:正交态自动求值
SymPy 内置的OrthogonalKet就实现了这一机制(sympy/physics/quantum/state.py#L628-L645):标签相同的正交态内积为 1,标签不同的为 0,无法判定时保留未求值形式:
>>> from sympy.physics.quantum import OrthogonalBra, OrthogonalKet >>> from sympy.abc import m, n >>> (OrthogonalBra(n)*OrthogonalKet(n)).doit() 1 >>> (OrthogonalBra(n)*OrthogonalKet(n+1)).doit() 0 >>> (OrthogonalBra(n)*OrthogonalKet(m)).doit() <n|m>与 Dagger 的交互:共轭与厄米共轭
内积的共轭由_eval_conjugate实现(innerproduct.py#L94-L95):
def _eval_conjugate(self): return InnerProduct(Dagger(self.ket), Dagger(self.bra))即 $\langle b|k\rangle^* = \langle k|b\rangle$:Bra 与 Ket 互换位置。这一规则与Dagger的通用行为一致——在 sympy/physics/quantum/dagger.py#L46-L48 的文档示例中:
>>> from sympy.physics.quantum import InnerProduct >>> Dagger(InnerProduct(Bra('a'), Ket('b'))) <b|a>配套测试test_innerproduct_dagger(sympy/physics/quantum/tests/test_innerproduct.py#L21-L25)对此进行了断言:Dagger(ip) == Dagger(k)*Dagger(b)。
多后端打印输出
InnerProduct针对不同打印后端实现了专用方法:
| 方法 | 输出风格 | 实现位置 |
|---|---|---|
_sympystr | 字符串:<b|k>,通过拼接 Bra 去掉右括号、Ket 去掉左括号生成 | innerproduct.py#L101-L104 |
_pretty | 美式打印(pretty printer),调用 Bra/Ket 的_print_contents_pretty与_pretty_brackets,按最高内容高度对齐尖括号 | innerproduct.py#L106-L120 |
_latex | LaTeX 输出:\left\langle ... \right. ...形式 | innerproduct.py#L122-L125 |
_sympyrepr | 可重建的 repr:InnerProduct(<b|,|k>) | innerproduct.py#L97-L99 |
其中_pretty方法复用了 sympy/physics/quantum/state.py#L142-L182 中_pretty_brackets的括号渲染逻辑:当状态内容高度超过一行时,会自动生成可伸缩的多行尖括号(支持 ASCII 与 Unicode 两种模式),保证复杂标签下的排版美观。
测试验证:行为契约一览
模块的单元测试集中在 sympy/physics/quantum/tests/test_innerproduct.py,可作为行为契约参考:
test_innerproduct:验证InnerProduct(b, k)的构造、bra/ket属性、b*k等价性、subs替换(InnerProduct(b, k).subs(b, Dagger(k)) == Dagger(k)*k);test_innerproduct_dagger:验证厄米共轭规则;test_doit:定义自定义状态类验证求值分发——FooKet通过_eval_innerproduct_FooBra返回Integer(1)、_eval_innerproduct_BarBra返回I,因此InnerProduct(BarBra('bar'), FooKet('foo')).doit() == I,且InnerProduct(Dagger(f), f).doit() == Integer(1)、InnerProduct(Dagger(f), Dagger(b)).doit() == -I(后者经共轭回退路径计算)。
实战:自定义状态类扩展内积求值
当内置状态无法满足需求时,可以像测试用例那样自定义量子态,为特定 Bra 类型注入内积计算规则:
from sympy.physics.quantum.state import StateBase, Ket, Bra from sympy.core.numbers import Integer class FooState(StateBase): pass class FooKet(Ket, FooState): @classmethod def dual_class(self): return FooBra def _eval_innerproduct_FooBra(self, bra): return Integer(1) # <foo|foo> 约定为 1 class FooBra(Bra, FooState): @classmethod def dual_class(self): return FooKet定义好_eval_innerproduct_<Bra类名>方法后,doit()会经由dispatch_method自动分发到该方法,将未求值的内积化简为具体数值。这是扩展 SymPy 量子模块内积计算能力、接入自定义正交基或波函数体系的标准做法。
InnerProduct由 sympy/physics/quantum/init.py#L48 导出,用户可通过from sympy.physics.quantum import InnerProduct直接导入。该模块对应的完整文档入口为 doc/src/modules/physics/quantum/innerproduct.rst,其内容由automodule指令从源码 docstring 自动生成,本文所涉及的所有用法示例均来自该模块的官方 docstring 与配套测试。
【免费下载链接】sympyA computer algebra system written in pure Python项目地址: https://gitcode.com/GitHub_Trending/sy/sympy
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考