Ratakan

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

Lapisan yang rata.

Lapisan rata meratakan masukan saat diterapkan tanpa mempengaruhi ukuran batch.

  • Pernyataan

    public typealias TangentVector = EmptyTangentVector
  • Membuat lapisan rata.

    Pernyataan

    public init()
  • Mengembalikan keluaran yang diperoleh dari penerapan lapisan ke masukan yang diberikan.

    Pernyataan

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

    Parameter

    input

    Masukan ke lapisan.

    Nilai Pengembalian

    Outputnya.