assume

class cuequivariance.assume(

irrep_class: str | Type[Irrep] | None = None,

layout: IrrepsLayout | None = None,

)

assume 是一个上下文管理器或装饰器，用于为一个代码块或函数假定 irrep 类和布局。

示例

作为上下文管理器

>>> with cue.assume(cue.SO3, cue.mul_ir):
...     rep = cue.IrrepsAndLayout("2x1")
>>> rep.irreps
2x1
>>> rep.layout
(mul,irrep)

作为装饰器

>>> @cue.assume(cue.SO3, cue.mul_ir)
... def foo():
...     return cue.IrrepsAndLayout("2x1")
>>> assert foo() == rep