MPSMatrixNeuron(3) | MetalPerformanceShaders.framework | MPSMatrixNeuron(3) |
MPSMatrixNeuron
#import <MPSMatrixNeuron.h>
Inherits MPSMatrixUnaryKernel.
(void) - setNeuronType:parameterA:parameterB:parameterC:
(MPSCNNNeuronType) - neuronType
(float) - neuronParameterA
(float) - neuronParameterB
(float) - neuronParameterC
(void) - setNeuronToPReLUWithParametersA:
(nonnull instancetype) - initWithDevice:
(void) - encodeToCommandBuffer:inputMatrix:biasVector:resultMatrix:
(nullable instancetype) - initWithCoder:device:
(nonnull instancetype) - copyWithZone:device:
NSUInteger sourceNumberOfFeatureVectors
NSUInteger sourceInputFeatureChannels
double alpha
MPSMatrixNeuron.h MetalPerformanceShaders.framework
Copyright:
This depends on Metal.framework.
A neuron activation kernel that operates on matrices.
A MPSMatrixNeuron object computes:
y = neuron(alpha * x + bias)
y is the output matrix, x is the input matrix corresponding
to a collection of input vectors and bias is a vector which is broadcast
and accumulated to each row of the intermediate result.
alpha is a scale factor applied to the input.
neuron() defines the pointwise function that is applied to the intermediate result.
Note: This function computes the same result as MPSMatrixFullyConnected that has
unit weight matrix.
Make a copy of this kernel for a new device -
See also:
Parameters:
Returns:
Reimplemented from MPSKernel.
Encode a MPSMatrixNeuron object to a command buffer.
Parameters:
Encodes the operation to the specified command buffer. resultMatrix must be large enough to hold a MIN(sourceNumberOfFeatureVectors, inputMatrix.rows - sourceMatrixOrigin.x) x MIN(inputMatrix.columns - sourceMatrixOrigin.y, sourceInputFeatureChannels) array.
The bias vector must contain at least MIN(inputMatrix.columns - sourceMatrixOrigin.y, sourceInputFeatureChannels) elements.
NSSecureCoding compatability See MPSKernel::initWithCoder.
Parameters:
Returns:
Reimplemented from MPSKernel.
Standard init with default properties per filter type
Parameters:
Returns:
Reimplemented from MPSKernel.
Getter funtion for neuronType set using setNeuronType:parameterA:parameterB:parameterC method
Getter funtion for neuronType set using setNeuronType:parameterA:parameterB:parameterC method
Getter funtion for neuronType set using setNeuronType:parameterA:parameterB:parameterC method
Getter funtion for neuronType set using setNeuronType:parameterA:parameterB:parameterC method
Add per output value neuron parameters A for PReLu neuron activation functions.
This method sets the neuron to PReLU, zeros parameters A and B and sets the per output value neuron parameters A to an array containing a unique value of A for each output value.
If the neuron function is f(v,a,b), it will apply
resultMatrix(i, j) = f( input(i, j), A[j], B[j] ) where j in [0, sourceInputFeatureChannels]
See https://arxiv.org/pdf/1502.01852.pdf for details.
All other neuron types, where parameter A and parameter B are shared across output values must be set using -setNeuronType:parameterA:parameterB:
Parameters:
Specifies a neuron activation function to be used.
This method can be used to add a neuron activation funtion of given type with associated scalar parameters A, B, and C that are shared across all output values. Note that this method can only be used to specify neurons which are specified by three (or fewer) parameters shared across all output values (or channels, in CNN nomenclature). It is an error to call this method for neuron activation functions like MPSCNNNeuronTypePReLU, which require per-channel parameter values. For those kind of neuron activation functions, use appropriate setter functions. An MPSMatrixNeuron kernel is initialized with a default neuron function of MPSCNNNeuronTypeNone.
Parameters:
The scale factor to apply to the input. Specified in double precision. Will be converted to the appropriate precision in the implementation subject to rounding and/or clamping as necessary. Defaults to 1.0 at initialization time.
The input size to to use in the operation. This is equivalent to the number of columns in the primary (input array) source matrix to consider and the number of channels to produce for the output matrix. This property is modifiable and defaults to NSUIntegerMax. At encode time the larger of this property or the available input size is used. The value of NSUIntegerMax thus indicates that all available columns in the input array (beginning at sourceMatrixOrigin.y) should be considered. Defines also the number of output feature channels. Note: The value used in the operation will be MIN(inputMatrix.columns - sourceMatrixOrigin.y, sourceInputFeatureChannels)
The number of input vectors which make up the input array. This is equivalent to the number of rows to consider from the primary source matrix. This property is modifiable and defaults to NSUIntegerMax. At encode time the larger of this property or the available number of inputs is used. The value of NSUIntegerMax thus indicates that all available input rows (beginning at sourceMatrixOrigin.x) should be considered.
Generated automatically by Doxygen for MetalPerformanceShaders.framework from the source code.
Mon Jul 9 2018 | Version MetalPerformanceShaders-119.3 |