本文共 226 字，大约阅读时间需要 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)是在最后增加一个维度。
转载地址：http://rtrti.baihongyu.com/