python - 如何向 tf.layers.Conv3D 提供输入层？

Question

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

Answer 1

您可以按以下方式重写代码:

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

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