tf.nn.crelu（）的静态形状未定义

Question

目前（版本0.11.0）tf.nn.crelu（）未正确设置静态形状：

f = tf.random_normal([50, 5, 7, 10])
f2 = tf.nn.relu(f)
print(f2.get_shape().as_list()) # [50, 5, 7, 10]
f3 = tf.nn.crelu(f)
print(f3.get_shape().as_list()) # [None, None, None, None]

with tf.Session() as sess:
    print(tf.__version__)
    py_f2, py_f3 = sess.run([f2, f3])

这是以Github issue #5912

提交的

Answer 1

解决方法：

f = tf.random_normal([50, 5, 7, 10])
[b, nx, ny, nz] = f.get_shape().as_list()
f3 = tf.nn.crelu(f)
f3.set_shape([b, nx, ny, 2*nz]
print(f3.get_shape().as_list()) # [50, 5, 7, 20]