@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.Предварительное условие
Количество элементов смещения должно делиться на группы.Предварительное условие
Смещение и масштаб должны иметь одинаковую форму.Декларация
Параметры
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.
Возвращает выходные данные, полученные в результате применения слоя к заданным входным данным.
Предварительное условие
Ось не может быть пакетной осью.Предварительное условие
Количество объектов входа и смещения должно быть одинаковым.Параметры
input
Входные данные слоя.
Возвращаемое значение
Вывод.