tensorflow中的函数

Posted on | Edited on | In | Views:
Symbols count in article: 279 | Reading time ≈ 1 mins.

tf.expand_dims

1
2
3
4
5
6
expand_dims(
    input,
    axis=None,
    name=None,
    dim=None
)

在 tensorflow/python/ops/array_ops.py中定义.

比较常用的情况是, 可以对一些训练图片增加一个batch的维度, 即从一个RGB图片[height, width, color_channel]变成 [1, height, width, color_channel], 这样图片就从一个普通图片tensor变成包含batch index的tensor.