Đổi hình dạng

@frozen
public struct Reshape<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPoint

Một lớp định hình lại.

  • Tuyên ngôn

    public typealias TangentVector = EmptyTangentVector

  • Hình dạng mục tiêu.

    Tuyên ngôn

    @noDerivative
public var shape: Tensor<Int32>

  • Tạo một lớp định hình lại.

    Tuyên ngôn

    public init(shape: Tensor<Int32>)

    Thông số

    shape

    Hình dạng mục tiêu, được biểu thị bằng một tenxơ.

  • Tạo một lớp định hình lại.

    Tuyên ngôn

    public init(_ shape: TensorShape)

    Thông số

    shape

    Hình dạng mục tiêu.

  • Trả về kết quả thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

    @differentiable
public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>

    Thông số

    input

    Đầu vào của lớp.

    Giá trị trả về

    Đầu ra.