标签: tensorflow
我希望用3(或4)个矢量来模拟物理量。即每个值都是3/4浮点数,其中每个浮点数都没有任何意义。 (4矢量版本用于相对论方程)。这些矢量代表位置,动量,磁场强度等。
查看文档,似乎我最好的举措是实现一个新的DType,然后对它们进行新的操作(例如矢量产品,交叉产品),但它并不表示python dtype接口是意思是可扩展的。
这是正确的方法,还是我错过了什么?