tensorflow 中tf.expand_dims函数用法

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/wshk918/article/details/82916030

tf.expand_dims()(维度扩展)函数用法

tf.expand_dims(input,position,name)    

其中input就是要扩展的变量,position就是选择在那个位置上扩展维度,如果input原来是3维的,那么position有四个数字(0,1,2,3)可以选择可以理解为插空位扩展

猜你喜欢

转载自blog.csdn.net/wshk918/article/details/82916030