首页 > 解决方案 > 在创建具有盒装类型(例如整数)数组的张量与张量流中的基元数组时是否存在差异

问题描述

我试图得到一个明确的答案,分别从Tensor.create(new int[3])和创建的两个张量之间是否存在任何差异Tensor.create(new Integer[3])。它们具有完全相同的数据类型 (INT32) 和形状 ([3])。我认为它们是相同的。但是,类setValue内部的调用Tensor是本机调用,我不能确定本机代码内部是否有任何区别。

标签: javatensorflow

解决方案


看起来一切都在此方法中被解析为原始类型,所以我猜 JNI 做了某种原始拆箱?我不太确定它是如何工作的。

但仅供参考,所有那些使用反射魔法的张量创建者都已在下一代 TensorFlow Java 中被删除,因为它们不具备高性能。您可能想查看此存储库以获取更多详细信息。


推荐阅读