MSELoss
public struct MSELoss : Loss
MSE loss. Currently it does reduce mean.
-
Declaration
Swift
public enum ReduceOp
-
Declaration
Swift
public var reduceOp: ReduceOp
-
Declaration
Swift
public init(_ reduceOp: ReduceOp = .mean)
-
Declaration
Swift
public func callAsFunction<T: DynamicGraph.AnyTensorGroup, U: DynamicGraph.AnyTensorGroup>( _ input: T, target: U, streamContext: StreamContext? ) -> [T.AnyTensor] where T.AnyTensor == U.AnyTensor