Neural network operation method and device, chip, electronic equipment and storage medium
The invention relates to the field of data calculation, and relates to a neural network operation method and device, a chip, electronic equipment and a storage medium. The neural network operation method comprises the following steps: acquiring input data of neural network operation and Wk * Hk sub-...
Gespeichert in:
Hauptverfasser: | , |
---|---|
Format: | Patent |
Sprache: | chi ; eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | The invention relates to the field of data calculation, and relates to a neural network operation method and device, a chip, electronic equipment and a storage medium. The neural network operation method comprises the following steps: acquiring input data of neural network operation and Wk * Hk sub-convolution kernel groups, and entering an operation step; n Wk * Hk * C convolution kernels of neural network operation are split to obtain N * Wk * Hk 1 * 1 * C sub convolution kernels, and the N * Wk * Hk 1 * 1 * C sub convolution kernels are divided into Wk * Hk sub convolution kernel groups; the operation step comprises the following steps of: rearranging input data according to a data rearrangement mode corresponding to each sub-convolution kernel group to obtain rearranged input data corresponding to each sub-convolution kernel group; performing convolution on each sub-convolution kernel group and the rearranged input data corresponding to each sub-convolution kernel group to obtain a convolution result of e |
---|