1
je code suivant:tensorflow: Comment obtenir sous-matrice pour chaque ligne de tenseur
import numpy as np
import tensorflow as tf
series = tf.placeholder(tf.float32, shape=[None, 5])
series_length = tf.placeholder(tf.int32, shape=[None])
useful_series = tf.magic_slice_function(series, series_length)
with tf.Session() as sess:
input_x = np.array([[1, 2, 3, 0, 0],
[2, 3, 0, 0, 0],
[1, 0, 0, 0, 0]])
input_y = np.array([[3], [2], [1]])
print(sess.run(useful_series, feed_dict={series: input_x, series_length: input_y}))
sortie prévue comme suit
[[1,2,3], [2,3], [1]]
J'ai essayé plusieurs fonctions, etc tf.gather, tf.slice. Tous ne fonctionnent pas. Qu'est-ce que le magic_slice_function?
Probablement, vous devez le faire en dehors tensorflow, puisque ce que vous voulez obtenir est pas un Tensor. –