Remodeler

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

Un calque de remodelage.

  • Déclaration

    public typealias TangentVector = EmptyTangentVector
  • La forme cible.

    Déclaration

    @noDerivative
    public var shape: Tensor<Int32>
  • Crée un calque de remodelage.

    Déclaration

    public init(shape: Tensor<Int32>)

    Paramètres

    shape

    La forme cible, représentée par un tenseur.

  • Crée un calque de remodelage.

    Déclaration

    public init(_ shape: TensorShape)

    Paramètres

    shape

    La forme cible.

  • Renvoie le résultat obtenu en appliquant le calque à l’entrée donnée.

    Déclaration

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

    Paramètres

    input

    L'entrée dans le calque.

    Valeur de retour

    Le résultat.