output = [[1,2,3],
[4,5,6],
[2, 5, 6]]
print(output[-1])
以这个为例,输出output[-1]的值,
[2, 5, 6]是结果,
则表示,下标为-1表示输出数组的最后一行数据值
用法:当有时候弄不清数组的最后一组数据的时候,可以用这个方法。
例如
def lstm(inputs):
cell = tf.nn.rnn_cell.BasicLSTMCell(num_units=OUTPUT_SIZE)
h0 = cell.zero_state(batch_size=tf.shape(inputs)[1],dtype=tf.float32)
#(timesteps,batchsize,dim)第二个才是batchsize
outputs,final_state = tf.nn.dynamic_rnn(cell,inputs = inputs,initial_state=h0,time_major=True)
return outputs[-1]
这里想输出lstm的最后一层输出的数据,直接写-1方便一些。