gpt4 book ai didi

keras - 如何 reshape keras中的张量?

转载 作者:行者123 更新时间:2023-12-01 11:17:03 30 4
gpt4 key购买 nike

我想 reshape 一个张量,它的形状是 (?,12,12,5,512) 到 (?,12,12,2560) 张量的形状。
有没有人可以帮助我?
我的代码如下。

conv5_1 = Conv3D(512, (3, 3, 3), activation='relu', padding='same')(drop4_1) # conv5_1: Tensor("conv3d_10/Relu:0", shape=(", 12, 12, 5, 512), dtype=float32)
conv5_1 = Conv3D(512, (3, 3, 3), activation='relu', padding='same')(conv5_1)
drop5_1 = Dropout(0.2)(conv5_1) # drop5_1: Tensor("dropout_8/cond/Merge:0", shape=(", 12, 12, 5, 512), dtype=float32)

我想在 drop5_1 之后制作 (?, 12, 12, 2560) 张量的形状。
谢谢

最佳答案

keras.layers.core.Reshape()功能很有帮助(另请参阅 document )。

reshaped = Reshape((12, 12, 2560))(drop5_1)

关于keras - 如何 reshape keras中的张量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49351038/

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