我有以下代码:
_X = np.arange(1, 7).reshape((2, 3))
_Y = np.arange(1, 7).reshape((3, 2))
X = tf.convert_to_tensor(_X)
Y = tf.convert_to_tensor(_Y)
# Matrix multiplication
out1 = tf.matmul(X, Y)
为此,我得到了这个错误:
TypeError: Value passed to parameter 'a' has DataType int64 not in list of allowed values: float16, float32, float64, int32, complex64, complex128
我正在使用最新版本的Tensorflow。有什么问题吗?
对tf.matmul的输入只接受这些数据类型:
将X和Y的数据类型更改为上述数据类型有效。
相关问题 更多 >
编程相关推荐