Skip to content

Commit

Permalink
Update variable name to reflect documentation description
Browse files Browse the repository at this point in the history
  • Loading branch information
jon-tow committed Jun 25, 2019
1 parent ada4daf commit bd7277e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Sources/TensorFlow/Layers/Cropping.swift
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,9 @@ public struct Cropping1D<Scalar: TensorFlowFloatingPoint>: Layer {
/// - Returns: The cropped 3D tensor of shape `[batch, cropped axis, feature]`.
@differentiable
public func callAsFunction(_ input: Tensor<Scalar>) -> Tensor<Scalar> {
let (start, end) = cropping
let (begin, end) = cropping
return input.slice(
lowerBounds: [0, start, 0],
lowerBounds: [0, begin, 0],
upperBounds: [input.shape[0], input.shape[1] - end, input.shape[2]])
}
}
Expand Down

0 comments on commit bd7277e

Please sign in to comment.