基于java打造的深度学习框架，助你快速搭建神经网络，实现模型推理与训练，引擎支持自动求导，多线程与GPU运算GPU支持CUDA

package com.omega.common.utils; import java.util.Vector; import org.ejml.simple.SimpleMatrix; import com.omega.common.task.Task; import com.omega.common.task.TaskEngine; import com.omega.engine.gpu.GPUOP; import com.omega.engine.pooling.PoolingType; /** * * @ClassName: MatrixOperation * * @author lijiaming * * @date 2020年7月24日 * * @Description: * TODO(用一句话描述该文件做什么) * */ public class MatrixOperation { private static final int threadNum = 8; /** * * @Title: exp * * @param x * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[] exp(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = (float)Math.exp(x[i]); } return temp; } public static float[] log(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { if(x[i] == 0) { temp[i] = (float)Math.log(1e-47); }else { temp[i] = (float)Math.log(x[i]); } } return temp; } public static float[] sin(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = (float)Math.sin(x[i]); } return temp; } public static float[] cos(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = (float)Math.cos(x[i]); } return temp; } public static float[] tan(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = (float)Math.tan(x[i]); } return temp; } public static float[] atan(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = (float)Math.atan(x[i]); } return temp; } public static float[] tan_back(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = (float)(1.0f / Math.pow(Math.cos(x[i]), 2)); } return temp; } public static float[] atan_back(float[] x) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = 1.0f / (1.0f + x[i] * x[i]); } return temp; } public static boolean isNaN(float[] x) { for(float v:x) { if(Float.isNaN(v)) { return true; } } return false; } public static boolean isInfinite(float[] x) { for(float v:x) { if(Float.isInfinite(v)) { return true; } } return false; } /** * * @Title: exp * * @param x * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[] pow(float[] x,float e) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = (float)Math.pow(x[i],e); } return temp; } /** * * @Title: exp * * @param x * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[][] pow(float[][] x,float e) { float[][] temp = MatrixUtils.zero(x.length,x[0].length); for(int i = 0;i<x.length;i++) { for(int j = 0;j<x[i].length;j++) { temp[i][j] = (float)Math.pow(x[i][j],e); } } return temp; } /** * * @Title: add * * @param x * @param b * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[] add(float[] x,float b) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = x[i] + b; } return temp; } public static float[] addToY(float[] x,float b,int n,int c,int h,int w,int[] position) { int dims = position[0]; int start= position[1]; int count = position[2]; switch (dims) { case 0: return addByNumberToY(x, b, n, c, h, w, start, count); case 1: return addByChannelToY(x, b, n, c, h, w, start, count); } return null; } public static float[] addByNumberToY(float[] x,float b,int n,int c,int h,int w,int start,int count) { int size = c * h * w; float[] y = MatrixUtils.clone(x); for(int i = 0;i<count;i++) { int tn = i / size + start; int tc = (i / h / w) % c; int th = i / w; int tw = i % h; int index = tn * size + tc * h * w + th * w + tw; y[index] = x[index] + b; } return y; } public static float[] addByChannelToY(float[] x,float b,int n,int c,int h,int w,int start,int count) { int size = c * h * w; int bc = count / n / h / w; float[] y = MatrixUtils.clone(x); for(int i = 0;i<count;i++) { int tn = i / size; int tc = (i / h / w) % bc + start; int th = i / w; int tw = i % h; int index = tn * size + tc * h * w + th * w + tw; y[index] = x[index] + b; } return y; } /** * * @Title: add * * @param x * @param b * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[][] add(float[][] x,float b) { if(x == null) { new RuntimeException("matrix is null."); } float[][] temp = MatrixUtils.zero(x.length,x[0].length); for(int i = 0;i<x.length;i++) { for(int j = 0;j<x[i].length;j++) { temp[i][j] = x[i][j] + b; } } return temp; } /** * * @Title: add * * @param x * @param b * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[][][] add(float[][][] x,float b) { if(x == null) { new RuntimeException("matrix is null."); } float[][][] temp = MatrixUtils.zero(x.length,x[0].length,x[0][0].length); for(int i = 0;i<x.length;i++) { for(int j = 0;j<x[i].length;j++) { for(int n = 0;n<x[i][j].length;n++) { temp[i][j][n] = x[i][j][n] + b; } } } return temp; } /** * * @Title: add * * @param x * @param b * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[][][][] add(float[][][][] x,float b) { if(x == null) { new RuntimeException("matrix is null."); } float[][][][] temp = MatrixUtils.zero(x.length,x[0].length,x[0][0].length,x[0][0][0].length); for(int n = 0;n<x.length;n++) { for(int c = 0;c<x[n].length;c++) { for(int h = 0;h<x[n][c].length;h++) { for(int w = 0;w<x[n][c][h].length;w++) { temp[n][c][h][w] = x[n][c][h][w] + b; } } } } return temp; } /** * * @Title: add * * @param x * @param b * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static float[] add(float[] x,float[] b) { float[] temp = MatrixUtils.zero(x.length); for(int i = 0;i<x.length;i++) { temp[i] = x[i] + b[i]; } return temp; } /** * * @Title: add * * @param x * @param b * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static void plus(float[] x,float[] b) { for(int i = 0;i<x.length;i++) { x[i] = x[i] + b[i]; } } /** * * @Title: add * * @param x * @param b * @return * * @Description: * TODO(这里用一句话描述这个方法的作用) * * @throws */ public static void add(float[] x,float[] b,int n,int c,int h,int w,int[] position) { int dims = position[0]; int start= position[1]; switch (dims) { case 0: addByNumber(x, b, n, c, h, w, start); break; case 1: addByChannel(x, b, n, c, h, w, start); break; } } public static void addByNumber(float[] x,float[] b,int n,int c,int h,int w,int start) { int size = c * h * w; for(int i = 0;i<b.length;i++) { int tn = i / size + start; int tc = (i / h / w) % c; int th = (i / w) % h; int tw = i % h; int index = tn * size + tc * h * w + th * w + tw; x[index] = x[index] + b[i]; } } public static void addByChannel(float[] x,float[] b,int n,int c,int h,int w,int start) { int bc = b.length / n / h / w; int size = bc * h * w; for(int i = 0;i<b.lengt