Functional

Swift

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

Swift

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

Swift

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

Swift

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

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

Swift

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

Swift

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

Swift

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