Loss

public protocol Loss

A generic loss function protocol.


                    
                    
                    callAsFunction(_:target:streamContext:)

Declaration

Swift

func callAsFunction<T: DynamicGraph.AnyTensorGroup, U: DynamicGraph.AnyTensorGroup>(
  _ input: T, target: U, streamContext: StreamContext?
) -> [T.AnyTensor] where T.AnyTensor == U.AnyTensor


                    
                    
                    callAsFunction(_:target:)

Extension method

Declaration

Swift

public func callAsFunction<T: DynamicGraph.AnyTensorGroup, U: DynamicGraph.AnyTensorGroup>(
  _ input: T, target: U
) -> [T.AnyTensor] where T.AnyTensor == U.AnyTensor