Constructor TopK

TopK(string, string, string, int, bool, bool, string[])

Initializes and returns an instance of TopK layer.

Declaration

public TopK(string name, string input, string k, int axis, bool largest, bool sorted, string[] outputNames)

Parameters

Type	Name	Description
string	name	The name to use for the values tensor of the layer.
string	input	The name to use for the input tensor of the layer.
string	k	The name to use for the single value 1D tensor containing the number of elements to calculate.
int	axis	The axis along which to perform the top-K operation.
bool	largest	Whether to calculate the top-K largest elements. If this is false the layer calculates the top-K smallest elements.
bool	sorted	Whether to return the elements in sorted order in the output tensor.
string[]	outputNames	A two-element array containing the names to use for the values and indices output tensors of the layer respectively.