재구성

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

레이어 모양 변경.

  • 선언

    public typealias TangentVector = EmptyTangentVector

  • 대상 모양입니다.

    선언

    @noDerivative
public var shape: Tensor<Int32>

  • 모양 변경 레이어를 만듭니다.

    선언

    public init(shape: Tensor<Int32>)

    매개변수

    shape

    텐서로 표현되는 대상 모양입니다.

  • 모양 변경 레이어를 만듭니다.

    선언

    public init(_ shape: TensorShape)

    매개변수

    shape

    대상 모양입니다.

  • 주어진 입력에 레이어를 적용하여 얻은 출력을 반환합니다.

    선언

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

    매개변수

    input

    레이어에 대한 입력입니다.

    반환 값

    출력.