tf.layers具有用于3d卷积的两个类。 1)tf.layers.conv3d和2)tf.layers.Conv3D
根据文档,不建议使用tf.layers.conv3d。所以我开始使用tf.layers.Conv3D。
但是tf.layers.Conv3D没有参数来指定tf.layers.conv3d所在的输入层。
有人可以帮我弄清楚如何为tf.layers.Conv3D图层提供输入吗?
答案 0 :(得分:1)
您可以通过以下方式重写代码:
# Old
output = tf.conv3d(input, *other_args)
# New
output = tf.keras.layers.Conv3D(*other_args)(input)