Question

我有一个张力常数

x = tf.zeros((N * (T - n + 1), n, D))

我有一个tensorflow占位符：

X = tf.placeholder(tf.float32, shape=(None, None, n_in))

我想把一些有价值的X分配给x，我会这么做：

x[N * i:N * (i + 1), :, :] = X[:, i:i + n, :]

我如何在张量流中做到这一点？

Answer 1

我会使用numpy分配数组块，然后转换回tensorflow：

with tf.Session() as sess:   
    #some tf operations here       
    # ...

    x_np = np.array(sess.run(x))
    X_np = np.array(sess.run(X))

    #assign with numpy:
    x_np[N * i:N * (i + 1), :, :] = X_np[:, i:i + n, :]

    x_result = tf.convert_to_tensor(x_np)

如何将值分配给Tensorflow SubTensor？

1 个答案: