如何获得keras'；一'；形状可变？

在我的keras计算图中，我需要一个所有1的矩阵，其中向量的第二维度可能会根据其他变量

n

而改变有没有办法让

K.ones

接受一个变量形状？我目前有一个办法可以使用

K.ones\u like

，因为我碰巧有另一个形状合适的变量，但我很好奇是否有更好的办法

示例代码（如果没有

x

，我如何获得

y

）：

将numpy导入为np
将keras.backend作为K导入
x=K.variable（np.array（[[1,2,3]，
[4, 5, 6]]))
n=K.variable（3，dtype=np.int32）
#这对我的情况有效，但只是因为我碰巧有x
y=K.one_like（x）
打印（K.eval（y））
#以下两项都失败，并出现“ValueError:使用序列设置数组元素”
y=K.ones（shape=K.stack（[K.constant（2，dtype=np.int32），n]））
y=K.one（形状=（2，n））
打印（K.eval（y））