python广播机制应用

原因

遇到一个小问题：假如有一个数组shape = (320,320),我想复制5遍这个数组，变成（5,320,320）。
原先的做法是

for _ in range(5):
	out_array[_,:,:] = intput_array[:,:]

感觉写的太烦了

后来想起来python有个广播机制

改成如下代码：

	# [5,1,1] * [1,320,320]
	out_array = np.ones([5,1,1]) * np.expand_dims(input_array, axis=-1) 

参考链接

https://dd96eb00.wiz06.com/wapp/pages/view/share/s/3tBKI039Uh7E2uegkP1UKqfr01mGRw0we17I2ogidB292tFf