ArgMax

class plaidml.op.ArgMax(value, axis=-1)

Maximum of elements along an axis.

Builds a tensor whose elements are the maximum value on some axis of an input tensor.

Methods

__init__(value[, axis])	Operation constructor.
bind(bindings)	Builds an output variable dictionary for the operation.
function(*args, **kwargs)	Invokes an Operation as a function.
sole_output()