分类:tensorflow
2020
09-29
一般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]...
继续阅读 >
2020
09-29
tf.tile()应用于需要张量扩展的场景,具体说来就是:如果现有一个形状如[width,height]的张量,需要得到一个基于原张量的,形状如[batch_size,width,height]的张量,其中每一个batch的内容都和原张量一模一样。tf.tile使用方法如:tile(input,multiples,name=None)importtensorflowastfa=tf.constant([7,19])a1=tf.tile(a,multiples=[3])#第一个维度扩充3遍b=tf.constant([[4,5],[3,5]])b1=tf.tile(b,multiples...
继续阅读 >
我就废话不多说了,大家还是直接看代码吧!importtensorflowastfimportnumpyasnpinput=tf.constant(1,shape=(64,10,1),dtype=tf.float32,name='input')#shape=(batch,in_width,in_channels)w=tf.constant(3,shape=(3,1,32),dtype=tf.float32,name='w')#shape=(filter_width,in_channels,out_channels)conv1=tf.nn.conv1d(input,w,2,'VALID')#2为步长print(conv1.shape)#宽度计算(width-kernel_size+1)/strides,(10-3...
继续阅读 >
我就废话不多说了,大家还是直接看代码吧!model=keras.models.Sequential([#卷积层1keras.layers.Conv2D(32,kernel_size=5,strides=1,padding="same",data_format="channels_last",activation=tf.nn.relu,kernel_regularizer=keras.regularizers.l2(0.01)),#池化层1keras.layers.MaxPool2D(pool_size=2,strides=2,padding="same"),#卷积层2keras.layers.Conv2D(64,kernel_size=5,strides=1,padding="same",data_format="c...
继续阅读 >
2020
09-27
2020
09-27