欢迎访问宙启技术站
智能推送

使用Python随机生成的obtain_input_shape()函数示例代码

发布时间:2023-12-11 03:21:01

下面是一个使用Python随机生成的obtain_input_shape()函数示例代码:

import random

def obtain_input_shape():
    input_shape = []
    num_channels = random.randint(1, 4)
    for _ in range(num_channels):
        input_shape.append(random.randint(1, 100))
    return tuple(input_shape)

这个函数用于生成随机的输入形状(input shape)。函数首先随机生成一个介于1和4之间的通道数(num_channels),然后随机生成每个通道的大小(介于1和100之间)。最后返回一个元组,包含生成的输入形状。

下面是一个使用例子:

input_shape = obtain_input_shape()
print("生成的随机输入形状为:", input_shape)

运行这个例子会输出类似以下结果:

生成的随机输入形状为: (56, 74, 23)

这表示生成的随机输入形状为一个包含3个通道,每个通道大小分别为56、74、23的形状。

这个函数可以在各种需要随机生成输入形状的场景中使用,例如生成神经网络模型的输入层形状,或者生成卷积神经网络的输入图片大小等。