TensorFlow入门--张量运算进阶

第1关：在指定位置修改维度 - expand_dims方法与squeeze方法

本关任务：编写一个可以根据要求增加或减少张量维数的小程序。

# -*- coding: utf-8 -*-
import tensorflow as tf

# 根据action中参数的值修改tensor的形状
# tensor是一个TensorFlow中的张量
# action是一个TensorFlow中的一维张量 (向量