使用Python生成随机的obtain_input_shape()函数示例代码
发布时间:2023-12-11 03:23:02
下面是一个示例代码,演示如何使用Python生成随机的obtain_input_shape()函数:
import random
def obtain_input_shape():
# 随机生成输入形状(一个三元组,表示输入的高度、宽度和通道数)
height = random.randint(10, 100)
width = random.randint(10, 100)
channels = random.randint(1, 10)
return (height, width, channels)
# 使用示例
input_shape = obtain_input_shape()
print(f"生成的随机输入形状:{input_shape}")
在这个示例代码中,obtain_input_shape()函数使用了random模块,利用random.randint()函数生成随机的整数。在这个例子中,生成的随机输入形状的高度、宽度和通道数都限定在一定的范围内。你可以根据自己的需求调整这些范围。
使用例子中的input_shape变量存储了调用obtain_input_shape()函数生成的随机输入形状,在最后通过print()函数将这个随机输入形状打印出来。
你可以运行这段代码多次,每次都会生成不同的随机输入形状。
希望这个示例代码能够帮助到你!如果有任何问题,请随时向我提问。
