CategoricalCrossEntropyLoss Structure Reference


                    
                    
                    trim0

Declaration

Swift

public var trim0: Float


                    
                    
                    trim1

Declaration

Swift

public var trim1: Float


                    
                    
                    init(trim0:trim1:)

Declaration

Swift

public init(trim0: Float = 0, trim1: Float = 1)


                    
                    
                    callAsFunction(_:target:streamContext:)

Declaration

Swift

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