Zmień kształt

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

Warstwa zmiany kształtu.

  • Deklaracja

    public typealias TangentVector = EmptyTangentVector

  • Docelowy kształt.

    Deklaracja

    @noDerivative
public var shape: Tensor<Int32>

  • Tworzy warstwę zmiany kształtu.

    Deklaracja

    public init(shape: Tensor<Int32>)

    Parametry

    shape

    Kształt docelowy reprezentowany przez tensor.

  • Tworzy warstwę zmiany kształtu.

    Deklaracja

    public init(_ shape: TensorShape)

    Parametry

    shape

    Docelowy kształt.

  • Zwraca wynik uzyskany po zastosowaniu warstwy do podanych danych wejściowych.

    Deklaracja

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

    Parametry

    input

    Wejście do warstwy.

    Wartość zwracana

    Wyjście.