RunningMeanStd

public struct RunningMeanStd<TensorElement>

  • Declaration

    Swift

    public var mean: TensorElement

  • Declaration

    Swift

    public var variance: TensorElement

  • Declaration

    Swift

    public var count: Int

  • Declaration

    Swift

    public init(mean: TensorElement, variance: TensorElement)

  • Declaration

    Swift

    public mutating func update(_ data: [TensorElement])

  • Declaration

    Swift

    public func norm(_ input: TensorElement) -> TensorElement