不确定数组的大小,但是迭代存进去的tensor的shape是一样的,那么可以考虑使用下面的方法。
其实我的应用是想申请一个大小不固定的数组,然后每个循环往里面存一次数,最后用于绘制2维图像。
可以使用python内自带的list功能:
buf = []
或者
buf=list([])
生成一个空的list
然后
for i in range (200):
buf.append(xxx)
存好之后,使用:
buf = np.array(buf)
转化为array
其实list也可以直接显示:
plt.plot(x,buf) plt.show()