Hier ist eine Funktion zur Verwendung von top_k
, um die maximalen k Aktivierungen der Kanäle zu nehmen. Sie können es an Ihren Zweck anpassen:
usd tf.reshape (), tf.matrix_transpose (), tf.nn.top_k (sortierte = False) und das Argument 'data_format' in tf.nn.conv2d (), siehe Ссылка für Details
Tags und Links python tensorflow