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

使用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()函数将这个随机输入形状打印出来。

你可以运行这段代码多次,每次都会生成不同的随机输入形状。

希望这个示例代码能够帮助到你!如果有任何问题,请随时向我提问。