python - 在 numpy 形状参数中使用元组

Question

第一篇文章 :) 如果我做错了什么，请不要开枪!

有没有更简写的方法来定义下面的形状？它有效，但有点冗长且不动态。

def neural_net_image_input(image_shape):
    """
    Return a Tensor for a batch of image input
    : image_shape: Shape of the images (taken from CIFAR10)
    : return: Tensor for image input.
    """
    x = tf.placeholder(tf.float32, shape=[None, image_shape[0], image_shape[1],  image_shape[2]], name='x')
    return x

我在 SO 和其他网站上搜索了大约一个小时但没有成功。我最初确实尝试过，

shape = [None, image_shape]

但出现错误(我理解)

TypeError: int() argument must be a string, a bytes-like object or a number, not 'tuple'

那么有没有办法将元组更改为我的形状参数可以接受的形式？

Answer 1

使用元组加法:

shape=(None,)+image_shape
# or if you want to allow lists and other sequences for image_shape:
shape=(None,)+tuple(image_shape)

或在具有可迭代解包概括的最新 Python 版本上:

shape=(None, *image_shape)