tensorflow中矩陣維度變換

如果需要擴(kuò)展一個(N, 1)的向量為(N, M)的矩陣,可以用tf.tile()和tf.expand_dims()

import tensorflow as tf
print(tf.__version__)
a = tf.constant([1, 2, 3, 4], dtype=tf.float32)
a0 = tf.expand_dims(a, 0)
a1 = tf.tile(a0, [2, 1])
with tf.Session() as sess:
    print(sess.run(a))
    print(sess.run(a0))
    print(sess.run(a1))
=======
'1.13.1'
[1, 2, 3, 4]
[[1, 2, 3, 4]]
[[1, 2, 3, 4], [1, 2, 3, 4]]

ref:
直觀的理解tensorflow中的tf.tile()函數(shù)
TensorFlow中張量轉(zhuǎn)置操作tf.expand_dims用法

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容