tensorflow官方网站:https://www.tensorflow.org/
1.tf.squeeze
从tensor中删除指定或者所有大小是1的维度
如:
capsules为20x1x10x16x1的tensor
cap=tf.squeeze(capsules, axis=1)
则cap为20x10x16x1的tensor
tensorflow官方网站:https://www.tensorflow.org/
1.tf.squeeze
从tensor中删除指定或者所有大小是1的维度
如:
capsules为20x1x10x16x1的tensor
cap=tf.squeeze(capsules, axis=1)
则cap为20x10x16x1的tensor