NFC: Lint loss functions.

Author: rxwei

Sources/DeepLearning/Loss.swift

@@ -18,12 +18,14 @@ import TensorFlow
 
 @differentiable
 public func meanSquaredError<Scalar: TensorFlowFloatingPoint>(
-    predicted: Tensor<Scalar>, expected: Tensor<Scalar>) -> Tensor<Scalar> {
+    predicted: Tensor<Scalar>, expected: Tensor<Scalar>
+) -> Tensor<Scalar> {
     return (expected - predicted).squared().mean()
 }
 
 @differentiable
 public func softmaxCrossEntropy<Scalar: TensorFlowFloatingPoint>(
-    logits: Tensor<Scalar>, labels: Tensor<Scalar>) -> Tensor<Scalar> {
+    logits: Tensor<Scalar>, labels: Tensor<Scalar>
+) -> Tensor<Scalar> {
     return -(labels * logSoftmax(logits)).mean(alongAxes: 0).sum()
 }