更新日志#
最新更改#
0.2.0 (2025-01-24)#
重大更改#
所有软件包的最低 Python 版本现在是 3.10。
cuet.TensorProduct和cuet.EquivariantTensorProduct现在要求输入形状为(batch_size, dim)或(1, dim)。不再允许维度为(dim,)的输入。cuex.IrrepsArray是cuex.RepArray的别名。cuex.RepArray.irreps和cuex.RepArray.segments不再是函数。现在是属性。cuex.IrrepsArray.is_simple已被cuex.RepArray.is_irreps_array替换。函数
cuet.spherical_harmonics已被 Torch 模块cuet.SphericalHarmonics替换。这样做是为了允许使用torch.jit.script和torch.compile。
新增#
添加了对
torch.compile的实验性支持。已知问题:导出到 c++ 无法工作。添加了
cue.IrrepsAndLayout:一个简单的类,它继承自cue.Rep并包含cue.Irreps和cue.IrrepsLayout。添加了
cuex.RepArray,用于表示任何类型的表示的数组(不仅是像以前的cuex.IrrepsArray那样的不可约表示)。
修复#
在
cuet(cuequivariance_torch) 中添加了对空批次维度的支持。将
README.md和LICENSE移动到源代码分发包中。修复了
cue.SegmentedTensorProduct.flop_cost对于 1 个操作数的特殊情况。
改进#
在
cuet.SymmetricTensorProduct中不再有 0 度特殊情况。
0.1.0 (2024-11-18)#
cuEquivariance 的 Beta 版本已发布。