如何检查python 3中tf.tensor数据集对象中存储的numpy.int64值的等效性?

bvpmtnay  于 2021-09-08  发布在  Java
关注(0)|答案(1)|浏览(342)

我想知道是否有一种简单的方法来检查类'numpy.int64'的tf.tensor数据集对象中存储的int32值的等价性。整数值会随着时间的推移而变化,我想在值为2时得到一个逻辑指示符'true',在值为2时得到'false'。我正在使用python 3和tensorflow 2.5.0。
这里有一些细节。当我打电话时:

print(class(token))

在我的dataset对象(名为“令牌”)上,我看到:

tf.Tensor([[2]], shape=(1, 1), dtype=int32)
<class 'numpy.int64'>

我想检查“token”中存储的整数值是否等于2,并在值相对于参考值2发生变化时接收逻辑指示符“true”或“false”。这似乎很容易,但无论我尝试何种索引或函数调用,我似乎都无法访问“token”中存储的整数值。我该怎么做?
当我打电话时:

print(token)

我懂了:

0

当我尝试时:

print(tf.math.equal(token,[[2]]))

我懂了:

tf.Tensor([[False]], shape=(1, 1), dtype=bool)

非常感谢。

xtfmy6hx

xtfmy6hx1#

运行这个

tf.squeeze(tf.math.equal(token,[[2]])).numpy()

这应该是输出 True .

相关问题