Remodelar

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

Uma camada de remodelagem.

  • Declaração

    public typealias TangentVector = EmptyTangentVector

  • A forma do alvo.

    Declaração

    @noDerivative
public var shape: Tensor<Int32>

  • Cria uma camada de remodelagem.

    Declaração

    public init(shape: Tensor<Int32>)

    Parâmetros

    shape

    A forma do alvo, representada por um tensor.

  • Cria uma camada de remodelagem.

    Declaração

    public init(_ shape: TensorShape)

    Parâmetros

    shape

    A forma do alvo.

  • Retorna a saída obtida da aplicação da camada à entrada fornecida.

    Declaração

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

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    A saída.