Como encontrar os valores máximos de k em um tensor 2-D no fluxo tensorial
Existe uma maneira de encontrar o topok
valores em um tensor 2-D em Tensorflow?
Eu posso usartf.nn.top_k
para um tensor 1-D, mas não pode funcionar com um tensor 2-D. Eu tenho um tensor 2-D com tamanho desconhecido, existe uma maneira de encontrar o topok
valores e seus índices?
Muito obrigado.