Python tf.raw_ops.ExpandDims用法及代碼示例

將維度 1 插入到張量的形狀中。

用法

tf.raw_ops.ExpandDims(
    input, axis, name=None
)

給定一個張量 input ，此操作在 input 的形狀的維度索引 axis 處插入維度 1。維度索引axis 從零開始；如果您為 axis 指定負數，則它從末尾倒數。

如果您想將批量維度添加到單個元素，此操作很有用。例如，如果您有一個形狀為 [height, width, channels] 的圖像，則可以使用 expand_dims(image, 0) 將其製作為一組 1 張圖像，這將生成形狀為 [1, height, width, channels] 。

其他示例：

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [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]

此操作要求：

-1-input.dims() <= dim <= input.dims()

此操作與 squeeze() 相關，它刪除大小為 1 的維度。