Remodelar

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

Una capa remodelada.

  • Declaración

    public typealias TangentVector = EmptyTangentVector

  • La forma del objetivo.

    Declaración

    @noDerivative
public var shape: Tensor<Int32>

  • Crea una capa de remodelación.

    Declaración

    public init(shape: Tensor<Int32>)

    Parámetros

    shape

    La forma objetivo, representada por un tensor.

  • Crea una capa de remodelación.

    Declaración

    public init(_ shape: TensorShape)

    Parámetros

    shape

    La forma del objetivo.

  • Devuelve la salida obtenida al aplicar la capa a la entrada dada.

    Declaración

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

    Parámetros

    input

    La entrada a la capa.

    Valor devuelto

    La salida.