/// <inheritdoc />
protected override bool _RowWiseOptimised(ref CudaFloat32NDArray array, Func <INDArray, INDArray> function)
{
if (function == SoftMax)
{
CudaFloat32NDArray internalArray = InternaliseArray(array);
array = CreateArrayFromHandle(DNDArray.CustomOp(internalArray.Handle, new CudaFloat32BackendHandle.CustomOpHandle(CudaFloat32BackendHandle.CustomOpType.RowWiseSoftmax)));
return(true);
}
return(false);
}
