/// <inheritdoc/>
Tensor IOps.ArgMax(Tensor X, int axis)
{
var O = m_Ops.ArgMax(X, axis);
LogLayerSummary(X.shape + " .+ " + O.shape);
LogOutputTensorSummary(O, Prefix + "ArgMax");
return O;
}
/// <inheritdoc/>
Tensor IOps.ArgMax(Tensor X, int axis)
{
var O = m_Ops.ArgMax(X, axis);
Reduce(X, O);
return(O);
}
/// <inheritdoc/>
Tensor IOps.ArgMax(Tensor X, int axis)
{
var Y = m_Ops1.ArgMax(X, axis);
var Z = m_Ops2.ArgMax(X, axis);
CheckSame(Y, Z, Layer.Type.ArgMax);
return(Y);
}
/// <inheritdoc/>
Tensor IOps.ArgMax(Tensor X, int axis)
{
var O = m_Ops.ArgMax(X, axis);
D.Log(X.shape + " .+ " + O.shape);
O.PrintDataPart(32, Prefix + "ArgMax");
return(O);
}
/// <inheritdoc/>
Tensor IOps.ArgMax(Tensor X, int axis)
{
var O = m_Ops.ArgMax(X, axis);
Reduce(X, O);
RegisterLayerStats();
return(O);
}