Functional Enumeration Reference


                    
                    
                    sum(_:streamContext:)

Element-wise addition

Declaration

Swift

public static func sum<T: DynamicGraph.TensorGroup>(
  _ inputs: [T], streamContext: StreamContext? = nil
) -> T


                    
                    
                    sum(_:streamContext:)

Element-wise addition

Declaration

Swift

public static func sum<T: DynamicGraph.TensorGroup>(
  _ inputs: T..., streamContext: StreamContext? = nil
) -> T


                    
                    
                    mul(left:right:scalar:streamContext:)

Broadcast element-wise multiplication

Declaration

Swift

public static func mul<T: DynamicGraph.TensorGroup>(
  left: T, right: T, scalar: Float32 = 1, streamContext: StreamContext? = nil
) -> T


                    
                    
                    add(left:right:leftScalar:rightScalar:streamContext:)

Broadcast element-wise addition

Declaration

Swift

public static func add<T: DynamicGraph.TensorGroup>(
  left: T, right: T, leftScalar: Float32 = 1, rightScalar: Float32 = 1,
  streamContext: StreamContext? = nil
) -> T


                    
                    
                    div(left:right:streamContext:)

Element-wise division

Declaration

Swift

public static func div<T: DynamicGraph.TensorGroup>(
  left: T, right: T, streamContext: StreamContext? = nil
) -> T


                    
                    
                    reciprocal(_:streamContext:)

Element-wise reciprocal

Declaration

Swift

public static func reciprocal<T: DynamicGraph.TensorGroup>(
  _ one: T, streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    log(_:streamContext:)

Element-wise log

Declaration

Swift

public static func log<T: DynamicGraph.TensorGroup>(_ one: T, streamContext: StreamContext? = nil)
  -> T


                    
                    
                    exp(_:streamContext:)

Element-wise exp

Declaration

Swift

public static func exp<T: DynamicGraph.TensorGroup>(_ one: T, streamContext: StreamContext? = nil)
  -> T


                    
                    
                    squareRoot(_:streamContext:)

Element-wise square root.

Declaration

Swift

public static func squareRoot<T: DynamicGraph.TensorGroup>(
  _ one: T, streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    softmax(_:streamContext:)

Softmax activation

Declaration

Swift

public static func softmax<T: DynamicGraph.TensorGroup>(
  _ one: T, streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    ReLU(_:streamContext:)

ReLU activation

Declaration

Swift

public static func ReLU<T: DynamicGraph.TensorGroup>(
  _ one: T, streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    sigmoid(_:streamContext:)

Sigmoid activation

Declaration

Swift

public static func sigmoid<T: DynamicGraph.TensorGroup>(
  _ one: T, streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    tanh(_:streamContext:)

Tanh activation

Declaration

Swift

public static func tanh<T: DynamicGraph.TensorGroup>(
  _ one: T, streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    swish(_:streamContext:)

Swish activation

Declaration

Swift

public static func swish<T: DynamicGraph.TensorGroup>(
  _ one: T, streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    averagePool(_:filterSize:hint:streamContext:)

Average pool

Declaration

Swift

public static func averagePool<T: DynamicGraph.TensorGroup>(
  _ one: T, filterSize: [Int], hint: Hint = Hint(), streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    maxPool(_:filterSize:hint:streamContext:)

Max pool

Declaration

Swift

public static func maxPool<T: DynamicGraph.TensorGroup>(
  _ one: T, filterSize: [Int], hint: Hint = Hint(), streamContext: StreamContext? = nil
)
  -> T


                    
                    
                    argmax(_:axis:streamContext:)

Argmax

Declaration

Swift

public static func argmax(
  _ one: DynamicGraph.AnyTensor, axis: Int, streamContext: StreamContext? = nil
)
  -> DynamicGraph.Tensor<Int32>


                    
                    
                    argmax(_:axis:streamContext:)

Argmax

Declaration

Swift

public static func argmax(
  _ one: DynamicGraph.Group<DynamicGraph.AnyTensor>, axis: Int,
  streamContext: StreamContext? = nil
)
  -> DynamicGraph.Group<DynamicGraph.Tensor<Int32>>


                    
                    
                    matmul(left:right:leftTranspose:rightTranspose:streamContext:)

Matrix multiplication

Declaration

Swift

public static func matmul<T: DynamicGraph.TensorGroup>(
  left: T, right: T, leftTranspose: (Int, Int) = (0, 0), rightTranspose: (Int, Int) = (0, 0),
  streamContext: StreamContext? = nil
) -> T


                    
                    
                    scalmul(left:right:streamContext:)

Scalar-matrix multiplication.

Declaration

Swift

public static func scalmul<T: DynamicGraph.TensorGroup>(
  left: Float, right: T, streamContext: StreamContext? = nil
) -> T


                    
                    
                    copy(from:to:streamContext:)

Make a copy.

Declaration

Swift

public static func copy<T: DynamicGraph.TensorGroup>(
  from: T, to: T, streamContext: StreamContext? = nil
)


                    
                    
                    indexSelect(input:index:streamContext:)

Select input tensor with another index tensor.

Declaration

Swift

public static func indexSelect<T: DynamicGraph.TensorGroup, U: DynamicGraph.TensorGroup>(
  input: T, index: U, streamContext: StreamContext? = nil
) -> T where U.ElementNumeric == Int32, T.AnyTensor == U.AnyTensor


                    
                    
                    indexSelect(input:index:streamContext:)

Select input tensor with another index tensor.

Declaration

Swift

public static func indexSelect<T: DynamicGraph.TensorGroup, U: DynamicGraph.TensorGroup>(
  input: T, index: U, streamContext: StreamContext? = nil
) -> T where U.ElementNumeric == Float32, T.AnyTensor == U.AnyTensor


                    
                    
                    maskedFill(input:mask:equalTo:fillWith:streamContext:)

Masked fill a tensor based on other tensor’s content equal to another.

Declaration

Swift

public static func maskedFill<T: DynamicGraph.TensorGroup, U: DynamicGraph.TensorGroup>(
  input: T, mask: U, equalTo: Float, fillWith: Float, streamContext: StreamContext? = nil
) -> T where T.AnyTensor == U.AnyTensor


                    
                    
                    min(_:_:streamContext:)

Element-wise min for two input tensors

Declaration

Swift

public static func min<T: DynamicGraph.TensorGroup>(
  _ left: T, _ right: T, streamContext: StreamContext? = nil
) -> T


                    
                    
                    max(_:_:streamContext:)

Element-wise max for two input tensors

Declaration

Swift

public static func max<T: DynamicGraph.TensorGroup>(
  _ left: T, _ right: T, streamContext: StreamContext? = nil
) -> T


                    
                    
                    min(_:_:)

Declaration

Swift

public static func min(_ left: Model.IO, _ right: Model.IO) -> Model.IO


                    
                    
                    max(_:_:)

Declaration

Swift

public static func max(_ left: Model.IO, _ right: Model.IO) -> Model.IO


                    
                    
                    concat(axis:_:)

Declaration

Swift

public static func concat(axis: Int, _ inputs: Model.IO...) -> Model.IO