在tensorflow中展开功能？

Question

Tensorflow中是否存在展开函数，其中矩阵转换为矢量。

示例：

矩阵[1 2 3; 4 5 6;]已展开＆＃39;到矢量[1 2 3 4 5 6]

查看数学运算，这似乎不可用：https://www.tensorflow.org/versions/r0.12/api_docs/python/math_ops.html

Answer 1

tf.reshape(a, shape=[-1])将＆＃34;展开＆＃34;使用行主顺序将张量a转换为向量。如果您想要不同的订单，可以先tf.transpose

import tensorflow as tf
a = tf.constant([[1, 2, 3], [4, 5, 6]])
b = tf.reshape(a, shape=[-1])
sess = tf.Session()
sess.run(b)  # => array([1, 2, 3, 4, 5, 6], dtype=int32)