กลุ่มนอร์ม

@frozen
public struct GroupNorm<Scalar> : Layer where Scalar : TensorFlowFloatingPoint

เลเยอร์ที่ใช้การทำให้เป็นมาตรฐานแบบกลุ่มกับอินพุตชุดย่อย

อ้างอิง: การทำให้เป็นมาตรฐานของกลุ่ม

  • ค่าออฟเซ็ตหรือที่เรียกว่าเบต้า

    คำประกาศ

    public var offset: Tensor<Scalar>
  • ค่าสเกลหรือที่เรียกว่าแกมมา

    คำประกาศ

    public var scale: Tensor<Scalar>
  • จำนวนกลุ่ม

    คำประกาศ

    @noDerivative
    public let groupCount: Int
  • แกนที่มีคุณลักษณะอยู่

    คำประกาศ

    @noDerivative
    public let axis: Int
  • ค่าความแปรปรวนเอปไซลอน

    คำประกาศ

    @noDerivative
    public let epsilon: Scalar
  • สร้างเลเยอร์การทำให้เป็นมาตรฐานของกลุ่ม

    เงื่อนไขเบื้องต้น

    แกนไม่สามารถเป็นแกนแบทช์ได้

    เงื่อนไขเบื้องต้น

    ออฟเซ็ตต้องมีอันดับ 1

    เงื่อนไขเบื้องต้น

    จำนวนองค์ประกอบของออฟเซ็ตต้องหารด้วยกลุ่ม

    เงื่อนไขเบื้องต้น

    ออฟเซ็ตและสเกลจะต้องมีรูปร่างเหมือนกัน

    คำประกาศ

    public init(
      offset: Tensor<Scalar>,
      scale: Tensor<Scalar>,
      groupCount: Int,
      axis: Int,
      epsilon: Scalar
    )

    พารามิเตอร์

    offset

    ค่าออฟเซ็ตเริ่มต้น

    scale

    ค่าสเกลเริ่มต้น

    groupCount

    จำนวนกลุ่ม

    axis

    แกนที่มีคุณลักษณะอยู่

    epsilon

    ค่าความแปรปรวนเอปไซลอน

  • สร้างเลเยอร์การทำให้เป็นมาตรฐานของกลุ่ม

    เงื่อนไขเบื้องต้น

    แกนไม่สามารถเป็นแกนแบทช์ได้

    เงื่อนไขเบื้องต้น

    จำนวนสถานที่ต้องหารด้วยกลุ่ม

    คำประกาศ

    public init(
      featureCount: Int,
      groupCount: Int,
      axis: Int = -1,
      epsilon: Scalar = 1e-3
    )

    พารามิเตอร์

    featureCount

    จำนวนคุณสมบัติ

    groupCount

    จำนวนกลุ่ม

    axis

    แกนที่มีคุณลักษณะอยู่ ค่าเริ่มต้นคือ -1

    epsilon

    สเกลาร์ขนาดเล็กบวกกับความแปรปรวน ค่าเริ่มต้นคือ 0.001

  • ส่งกลับผลลัพธ์ที่ได้จากการใช้เลเยอร์กับอินพุตที่กำหนด

    เงื่อนไขเบื้องต้น

    แกนไม่สามารถเป็นแกนแบทช์ได้

    เงื่อนไขเบื้องต้น

    จำนวนคุณลักษณะของอินพุตและออฟเซ็ตต้องเท่ากัน

    คำประกาศ

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

    พารามิเตอร์

    input

    อินพุตไปยังเลเยอร์

    ค่าส่งคืน

    ผลลัพธ์.