SymPy 量子力学模块的 InnerProduct:Bra-Ket 符号内积的实现原理与实战指南
2026/9/15 14:22:33 网站建设 项目流程

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" 一节),与CommutatorAntiCommutatorDaggerTensorProduct等量子运算类并列,是 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): ...

构造参数

参数类型含义
braBraBase或子类内积左侧的 Bra,即 $\langle bra \mid$
ketKetBase或子类内积右侧的 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>

braket属性分别返回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环节可以直接使用atomssubs
  • 保持"不存在可交换的 QExpr 子类"这一约束,简化了整个模块的设计。

因此类上设置了kind = NumberKindis_complex = True,与 QExpr 的is_commutative = False形成鲜明对比。这一性质解释了上节中内积因子在乘积中可任意换位、自动左移的行为。

doit():从符号内积到实际求值

InnerProduct默认是未求值的符号对象,真正计算内积数值/符号结果的方法是doit()(innerproduct.py#L127-L139)。其求值策略分两级:

  1. 首先尝试调用self.ket._eval_innerproduct(self.bra, **hints),即询问 Ket 是否知道如何与给定 Bra 计算内积;
  2. 若抛出NotImplementedError,则回退为对偶方案:计算conjugate(self.bra.dual._eval_innerproduct(self.ket.dual))(利用 $\langle b|k\rangle^* = \langle k|b\rangle$),再取共轭;
  3. 若两级都失败(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
_latexLaTeX 输出:\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),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询