tensorflow切片保留维度——采用冒号切片

img[:,:,0]则不会保留维度。——(32,32)
img[:,:,0:1]就能够保留维度。——(32,32,1)

总结,采用冒号切片方式即可保留维度。

Guess you like

Origin blog.csdn.net/qq_44065334/article/details/120589042