Cómo encontrar los valores de k superiores en un tensor 2-D en tensorflow
¿Hay alguna manera de encontrar el topk valores en un tensor 2-D en Tensorflow?
Puedo usartf.nn.top_k para un tensor 1-D pero no puede funcionar con un tensor 2-D. Tengo un tensor 2-D con un tamaño desconocido, ¿hay alguna manera de encontrar la parte superiork valores y sus índices?
Muchas gracias