Meilleur moyen pour aplatir un 2D tenseur contenant un vecteur dans TensorFlow?
Quel est le moyen le plus efficace pour aplatir un 2D tenseur qui est en fait à l'horizontale ou à la verticale du vecteur dans un 1D tenseur?
Est-il une différence en termes de performances entre:
tf.reshape(w, [-1])
et
tf.squeeze(w)
?
Vous devez vous connecter pour publier un commentaire.
Les deux
tf.remodeler(w, [-1])
ettf.squeeze(w)
sont "bon marché", en ce qu'elles ne fonctionnent que sur les métadonnées (c'est à dire la forme du tenseur donné, et ne modifient pas les données elles-mêmes. Des deuxtf.reshape()
a légèrement plus simple de la logique interne de la, mais les performances des deux devrait être identique.