Adding basic functionality to specify the feature axis of the input tensor...
Adding basic functionality to specify the feature axis of the input tensor (setting the default to -1)
Adding basic functionality to specify the feature axis of the input tensor (setting the default to -1)