ग्रुपनॉर्म

@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

    परत में इनपुट.

    प्रतिलाभ की मात्रा

    उत्पादन।