the Torch module update

[AlbertZyy]

fem.Integrators

Update

• 简化了 typehint
• 将 linear 和 bilinear 类型的数值积分计算，以及对 coef 形状的处理，封装为单独的函数，大幅减少积分子之间的代码重复。

New

• 加入可选的缓存机制，避免某些不变的积分在迭代中的重复计算。机制在启用后由模块自动完成，不改变用户的使用方式。
• 加入对批次数据的支持。现在 coef 和 source 可以接受第 0 个轴作为 batch 轴，输出最后一个轴是 batch 的张量。

fem.LinearForm & fem.BilinearForm

Update

• 修复了在没有添加任何 domain integrator 时（但是有 boundary integrator），assembly() 会出错的问题。现在即使未添加任何积分子，组装器仍可以输出一个空的（全零）稀疏张量。
• 在 LinearForm 的 assembly 方法中加入了 return_dense 参数并默认为 True。

New

• 加入对批次数据的支持（参数 batch_size），输出最后一个轴是 batch 的稀疏张量。

fem.DirichletBC

New

• 加入对批次右端向量的支持，此处自动适应。要求 batch dimension 在最后一个轴。