相同元组上的AssertionError（Tensorflow keras）

2024-04-26 • 问答

x = tf.keras.layers.Input((2,))
z = x.shape
y = (None,2)
print("z: ",z)
print("y: ",y)
assert z == y

输出：

z: (None,2)
y: (None,2)
Traceback (most recent call last):
  File "",line 113,in <module>
    assert z == y
AssertionError

当z和y的形状相同时，为什么会给出断言错误？

弄清楚了。元组有不同的类型。

print(type(z))
print(type(y))

Output: 
<class 'tensorflow.python.framework.tensor_shape.TensorShape'>
<class 'tuple'>

tensor.shape.as_list（）将为您提供列表格式的形状。

data

本文链接：https://www.f2er.com/3168602.html