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