function X = internal_softmax(X, channelDim)
% This function is undocumented and will change in a future release.

%   Copyright 2019 The MathWorks, Inc.

% Ensure logicals are cast to single
X = nnet.internal.cnnhost.castToSuperiorFloat(X);

X = nnet.internal.cnnhost.softmaxForward(X, channelDim);

end

