分类:指定列
我就废话不多说了,大家还是直接看代码吧~In[1]:importosIn[2]:os.environ["CUDA_VISIBLE_DEVICES"]="0"In[3]:importtensorflowastfIn[4]:sess=tf.Session()In[5]:input=tf.constant([[[1,2,3],[4,5,6],[7,8,9]],[[10,11,12],[13,14,15],[1...:6,17,18]]])In[6]:input.get_shape()Out[6]:TensorShape([Dimension(2),Dimension(3),Dimension(3)])In[7]:input_2=input[:,:,2]In[8]:print(sess.run...
继续阅读 >