cos_sim fsp gru lstm_unit match_matrix_tensor max_pool2d_with_index max_pool3d_with_index maxout pool2d pool3d reduce_max reduce_min reduce_prod reshape rnn_memory_helper sequence_softmax spp tensor_array_to_tensor transpose unsqueeze