OperatorFuncs

public class OperatorFuncs

Convenient function APIs of operators.


                    
                    
                    calculate(_:input:output:)

Undocumented

Declaration

Swift

internal static func calculate(_ op: inout ComputableOperator, input: [Tensor], output: Tensor?) -> Tensor

Show on GitHub

Unary operators


                    
                    
                    sinh(_:output:)

sinh

Declaration

Swift

public static func sinh(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    tan(_:output:)

tan

Declaration

Swift

public static func tan(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    tanh(_:output:)

tanh

Declaration

Swift

public static func tanh(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    sin(_:output:)

sin

Declaration

Swift

public static func sin(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    cos(_:output:)

cos

Declaration

Swift

public static func cos(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    cosh(_:output:)

cosh

Declaration

Swift

public static func cosh(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    arcsin(_:output:)

arcsin

Declaration

Swift

public static func arcsin(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    arcsinh(_:output:)

arcsinh

Declaration

Swift

public static func arcsinh(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    arccos(_:output:)

arccos

Declaration

Swift

public static func arccos(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    arccosh(_:output:)

arccosh

Declaration

Swift

public static func arccosh(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    arctan(_:output:)

arctan

Declaration

Swift

public static func arctan(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    arctanh(_:output:)

arctanh

Declaration

Swift

public static func arctanh(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    degree(_:output:)

degree

Declaration

Swift

public static func degree(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    radian(_:output:)

radian

Declaration

Swift

public static func radian(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    floor(_:output:)

floor

Declaration

Swift

public static func floor(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    ceil(_:output:)

ceil

Declaration

Swift

public static func ceil(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    rint(_:output:)

rint

Declaration

Swift

public static func rint(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    round(_:output:)

round

Declaration

Swift

public static func round(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    abs(_:output:)

abs

Declaration

Swift

public static func abs(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    square(_:output:)

square

Declaration

Swift

public static func square(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    sqrt(_:output:)

sqrt

Declaration

Swift

public static func sqrt(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    exp(_:output:)

exp

Declaration

Swift

public static func exp(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    expm1(_:output:)

expm1

Declaration

Swift

public static func expm1(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    rsqrt(_:output:)

rsqrt

Declaration

Swift

public static func rsqrt(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    log(_:output:)

log

Declaration

Swift

public static func log(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    log2(_:output:)

log2

Declaration

Swift

public static func log2(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    log10(_:output:)

log10

Declaration

Swift

public static func log10(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    log1p(_:output:)

log1p

Declaration

Swift

public static func log1p(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub

Binary operators


                    
                    
                    add(_:_:output:)

add

Declaration

Swift

public static func add(_ inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    sub(_:_:output:)

sub

Declaration

Swift

public static func sub(_ inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    mult(_:_:output:)

mult

Declaration

Swift

public static func mult(_ inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    div(_:_:output:)

div

Declaration

Swift

public static func div(_ inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    rdiv(_:_:output:)

rdiv

Declaration

Swift

public static func rdiv(_ inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    pow(_:_:output:)

pow

Declaration

Swift

public static func pow(_ inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub

Matrix operators


                    
                    
                    matrixMult(_:_:transposeA:transeposeB:output:)

Matrix multiplication

Declaration

Swift

public static func matrixMult(_ inputA: Tensor, _ inputB: Tensor,
                              transposeA:Bool = false, transeposeB: Bool = false,
                              output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    matrixTranspose(_:output:)

Matrix transpose

Declaration

Swift

public static func matrixTranspose(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub

Broadcast operators


                    
                    
                    broadcast(_:_:output:)

broadcast

Declaration

Swift

public static func broadcast(_ shape: TensorShape, _ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    add(broadcast:_:output:)

Broadcasting add

Declaration

Swift

public static func add(broadcast inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    sub(broadcast:_:output:)

Broadcasting sub

Declaration

Swift

public static func sub(broadcast inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    mult(broadcast:_:output:)

Broadcasting mult

Declaration

Swift

public static func mult(broadcast inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    div(broadcast:_:output:)

Broadcasting div

Declaration

Swift

public static func div(broadcast inputA: Tensor, _ inputB: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub

Reduce operators


                    
                    
                    reduceSum(_:axis:keepDim:output:)

Reduce sum

Declaration

Swift

public static func reduceSum(_ input: Tensor, axis: [Int], keepDim: Bool = false, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    reduceProduct(_:axis:keepDim:output:)

Reduce product

Declaration

Swift

public static func reduceProduct(_ input: Tensor, axis: [Int], keepDim: Bool = false, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    reduceMax(_:axis:keepDim:output:)

Reduce max

Declaration

Swift

public static func reduceMax(_ input: Tensor, axis: [Int], keepDim: Bool = false, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    reduceMin(_:axis:keepDim:output:)

Reduce min

Declaration

Swift

public static func reduceMin(_ input: Tensor, axis: [Int], keepDim: Bool = false, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    reduceMean(_:axis:keepDim:output:)

Reduce mean

Declaration

Swift

public static func reduceMean(_ input: Tensor, axis: [Int], keepDim: Bool = false, output: Tensor? = nil) -> Tensor

Show on GitHub

Activiation operators


                    
                    
                    relu(_:alpha:output:)

ReLU

Declaration

Swift

public static func relu(_ input: Tensor, alpha: Float = 0.0, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    sigmoid(_:output:)

sigmoid

Declaration

Swift

public static func sigmoid(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    softplus(_:output:)

softplus

Declaration

Swift

public static func softplus(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    softsign(_:output:)

softsign

Declaration

Swift

public static func softsign(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    linear(_:output:)

linear

Declaration

Swift

public static func linear(_ input: Tensor, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    elu(_:alpha:output:)

ELU

Declaration

Swift

public static func elu(_ input: Tensor, alpha: Float = 1.0, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    selu(_:alpha:scale:output:)

SELU

Declaration

Swift

public static func selu(_ input: Tensor, alpha: Float = 1.673263, scale: Float = 1.050701, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    softmax(_:dim:output:)

Softmax

Declaration

Swift

public static func softmax(_ input: Tensor, dim: Int = -1, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    leakyReLU(_:alpha:output:)

LeakyReLU

Declaration

Swift

public static func leakyReLU(_ input: Tensor, alpha: Float = 0.3, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    thresholdedReLU(_:alpha:output:)

ThresholdedReLU

Declaration

Swift

public static func thresholdedReLU(_ input: Tensor, alpha: Float = 1.0, output: Tensor? = nil) -> Tensor

Show on GitHub

nn operators


                    
                    
                    fc(_:numUnits:weight:bias:biasEnabled:output:)

FullyConnected

Declaration

Swift

public static func fc(_ input: Tensor, numUnits: Int,
                      weight: Tensor, bias: Tensor? = nil,
                      biasEnabled: Bool = true, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    batchnorm(_:movingMean:movingVar:channelOrder:useScale:scale:useOffset:offset:epsilon:output:)

batchnorm

Declaration

Swift

public static func batchnorm(_ input: Tensor, movingMean: Tensor, movingVar: Tensor,
                             channelOrder: TensorChannelOrder = TensorChannelOrder.Last,
                             useScale: Bool = true, scale: Tensor? = nil,
                             useOffset: Bool = true, offset: Tensor? = nil,
                             epsilon: Float = 0.001, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    maxPool2D(_:kernelSize:stride:channelOrder:paddingMode:output:)

MaxPool2D

Declaration

Swift

public static func maxPool2D(_ input: Tensor, kernelSize: [Int], stride: [Int]? = nil,
                             channelOrder: TensorChannelOrder = TensorChannelOrder.Last,
                             paddingMode: PaddingMode = PaddingMode.Valid, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    avgPool2D(_:kernelSize:stride:channelOrder:paddingMode:output:)

AvgPool2D

Declaration

Swift

public static func avgPool2D(_ input: Tensor, kernelSize: [Int], stride: [Int]? = nil,
                             channelOrder: TensorChannelOrder = TensorChannelOrder.Last,
                             paddingMode: PaddingMode = PaddingMode.Valid, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    sumPool2D(_:kernelSize:stride:channelOrder:paddingMode:output:)

SumPool2D

Declaration

Swift

public static func sumPool2D(_ input: Tensor, kernelSize: [Int], stride: [Int]? = nil,
                             channelOrder: TensorChannelOrder = TensorChannelOrder.Last,
                             paddingMode: PaddingMode = PaddingMode.Valid, output: Tensor? = nil) -> Tensor

Show on GitHub


                    
                    
                    conv2D(_:numFilters:kernelSize:weight:bias:biasEnabled:stride:padMode:channelPosition:output:)

convolution 2D

Declaration

Swift

public static func conv2D(_ input: Tensor, numFilters: Int, kernelSize: [Int],
                          weight: Tensor, bias: Tensor? = nil, biasEnabled: Bool = true,
                          stride: [Int] = [1, 1], padMode: PaddingMode = .Valid,
                          channelPosition: TensorChannelOrder = TensorChannelOrder.First,
                          output: Tensor? = nil) -> Tensor

Show on GitHub