如何在tensorflow中实现自定义数据类型

时间:2018-06-15 00:34:48

标签: tensorflow

我希望用3(或4)个矢量来模拟物理量。即每个值都是3/4浮点数,其中每个浮点数都没有任何意义。 (4矢量版本用于相对论方程)。这些矢量代表位置,动量,磁场强度等。

查看文档,似乎我最好的举措是实现一个新的DType,然后对它们进行新的操作(例如矢量产品,交叉产品),但它并不表示python dtype接口是意思是可扩展的。

这是正确的方法,还是我错过了什么?

0 个答案:

没有答案