federer/dist/src/server/utils/AverageWeights.d.ts

Experiments in asynchronous federated learning and decentralized learning

import { Weights } from "../../common";
/** Average of tensors. The average can be updated by adding tensors to it. */
export declare class AverageWeights {
    #private;
    /** Whether anything has been added to the average yet. */
    isDefined(): boolean;
    /** Add a tensor to the average. */
    add(tensor: Weights): void;
    /** Get the average value. */
    average(): Weights;
    /** Get the number of values that the average represents. */
    get count(): number;
    /** Dispose of the {@link AverageWeights}, discarding the memory backing it. */
    dispose(): void;
    /** Whether the {@link AverageWeights} has been disposed. */
    get isDisposed(): boolean;
}
//# sourceMappingURL=AverageWeights.d.ts.map