Functional
public enum Functional
Undocumented
-
Element-wise addition
Declaration
Swift
public static func sum<T: DynamicGraph.TensorGroup>( _ inputs: T..., streamContext: StreamContext? = nil ) -> T
-
Broadcast element-wise multiplication
Declaration
Swift
public static func mul<T: DynamicGraph.TensorGroup>( left: T, right: T, scalar: Float32 = 1, streamContext: StreamContext? = nil ) -> T
-
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
-
Element-wise log
Declaration
Swift
public static func log<T: DynamicGraph.TensorGroup>(_ one: T, streamContext: StreamContext? = nil) -> T
-
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
-
Scalar-matrix multiplication.
Declaration
Swift
public static func scalmul<T: DynamicGraph.TensorGroup>( left: Float, right: T, streamContext: StreamContext? = nil ) -> T
-
Make a copy.
Declaration
Swift
public static func copy<T: DynamicGraph.TensorGroup>( from: T, to: T, streamContext: StreamContext? = nil )
-
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