分类：维度扩展

2020
09-29

在TensorFlow中实现矩阵维度扩展

一般TensorFlow中扩展维度可以使用tf.expand_dims()。近来发现另一种可以直接运用取数据操作符[]就能扩展维度的方法。用法很简单，在要扩展的维度上加上tf.newaxis就行了。foo=tf.constant([[1,2,3],[4,5,6],[7,8,9]])print(foo[tf.newaxis,:,:].eval())#=>[[[1,2,3],[4,5,6],[7,8,9]]]print(foo[:,tf.newaxis,:].eval())#=>[[[1,2,3]],[[4,5,6]],[[7,8,9]]]print(foo[:,:,tf.newaxis].eval())#=>[[[1],[2]... 继续阅读 >