将矩阵在特定位置增加一维,举例如下:
如果# 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)是在最后增加一个维度。
将矩阵在特定位置增加一维,举例如下:
如果# 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)是在最后增加一个维度。