gpt4 book ai didi

python - 如何在 tensorflow 中将我的三个 2D 张量组合成一个 3D 张量?

转载 作者:太空宇宙 更新时间:2023-11-04 07:59:09 24 4
gpt4 key购买 nike

你好,我是 tensorflow 的新手,目前,我正在处理彩色图像,它是 PCAS。

我以“红色”、“绿色”和“蓝色”的形式提取了 PCAS,还计算了与“红色”、“绿色”和“蓝色”组件相关的权重。

完成上述所有操作后,我想将所有三个 2D 矩阵组合成一个 3D 矩阵。

对于 tensorflow ,它将是 3D 张量。

def multi(h0,ppca,mu,i,scope=None):

with tf.variable_scope(scope or"multi"):
return tf.matmul(ppca[:,:,0],h0[i,:,:,0]) + tf.reshape(mu[:,0],[4096,1]) , tf.matmul(ppca[:,:,1],h0[i,:,:,1]) + tf.reshape(mu[:,1],[4096,1]) ,tf.matmul(ppca[:,:,2],h0[i,:,:,2]) + tf.reshape(mu[:,2],[4096,1])

所以从上面的函数中,我将得到所有三个不同的 2D 张量,并想将这些 2D 张量组合成维度为 [4096,1,3] 的单个 3D 张量

我该怎么做?非常感谢任何帮助。

最佳答案

你需要像这样连接它们:

three_d_image = tf.concat(0, [[r], [g], [b]])

这告诉 tensorflow 沿 x 维度连接它们并将每个张量视为矩阵。

在 r、g、b 张量周围没有附加括号的情况下执行相同的操作将尝试将它们连接到一个大的二维矩阵

关于python - 如何在 tensorflow 中将我的三个 2D 张量组合成一个 3D 张量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43489012/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com