本文主要是介绍tf.expand_dims()用法,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
将矩阵在特定位置增加一维,举例如下:
如果# t2是一个形状为 [2, 3, 5]的张量,以下操作将使该张量的维度发生以下变化。
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
特别的,expand_dims(t2, -1)是在最后增加一个维度。
这篇关于tf.expand_dims()用法的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!