可以看到np.random.randint(1,10)的前两个参数是low和high,没有第三个参数size,所以返回值是一个int类型的数字。而line接受的参数是ndarray类型,所以会报错。应该np.random.randint(1,10,1)确保生成ndarray类型。

注:tensor类型也可以。还可以混用。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐