如何连接两个不同形状的张量

2023-12-07

我有两个张量,get_shape = [?, 400] 和 [?, 1176]。我想将它们连接成一个大小为 [?, 1576] 的张量。

我尝试了 concat,但它要求两者具有相同的维度。

怎么办呢?


希望您通过批量大小传递相同维度的输入。

import tensorflow as tf
import numpy as np

t1 = tf.placeholder(tf.float32, [None, 400])
t2 = tf.placeholder(tf.float32, [None, 1176])
t3 = tf.concat([t1, t2], axis = 1)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    t3_val = sess.run(t3, feed_dict = {t1: np.ones((300, 400)), t2: np.ones((300, 1176))})

    print(t3_val.shape)
    # (300, 1576)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何连接两个不同形状的张量 的相关文章

随机推荐