Data processing method and device for neural network

The invention discloses a data processing method and device of a neural network, and relates to the field of artificial intelligence. The method includes dynamically segmenting input data according to a data volume of the input data, a first feature of an internal memory in a chip running a neural n...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: XIONG LEJIN, YUAN HONGHUI, GAO SHANQING, GAO LIWEN
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:The invention discloses a data processing method and device of a neural network, and relates to the field of artificial intelligence. The method includes dynamically segmenting input data according to a data volume of the input data, a first feature of an internal memory in a chip running a neural network, and a second feature of a plurality of layers in the neural network, and setting different batch sizes for the layers in the neural network. Due to the fact that the reasonable batch size is set for each layer in the neural network, the internal memory can be fully utilized to store the interlayer data of the neural network in the reasoning process of the neural network, the utilization rate of the internal memory is improved, and the calculation efficiency of hardware for operating the neural network is ensured. 本申请公开了一种神经网络的数据处理方法及装置,涉及人工智能领域。所述方法包括依据输入数据的数据量、运行神经网络的芯片内的内部存储器的第一特征和神经网络中多个层的第二特征动态切分输入数据,为神经网络中的层设置不同的批大小。由于通过为神经网络中的每层设置合理的批大小,在神经网络推理过程中,可充分利用内部存储器存储神经网络的层间数据,从而提高了内部存储器的利用率,以及确保运行神经网络的硬件的计算效