如何为tf.layers.Conv3D提供输入层?

时间:2019-09-20 20:59:46

标签: python tensorflow

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图层提供输入吗?

1 个答案:

答案 0 :(得分:1)

您可以通过以下方式重写代码:

    # Old
    output = tf.conv3d(input, *other_args)

    # New
    output = tf.keras.layers.Conv3D(*other_args)(input)