org.dromara.jcudax

类 JCudax

java.lang.Object

org.dromara.jcudax.JCudax

public class JCudax
extends Object

构造器概要

构造器

构造器和说明
JCudax()

方法概要

限定符和类型	方法和说明
static void	initialize()
static void	matrixSoftMaxPd(double[] qkt, double[] errorMatrix, double[] grMatrix, int x, int y, double wordVectorDimension)
static void	matrixSoftMaxPdFp32(float[] qkt, float[] errorMatrix, float[] grMatrix, int x, int y, float wordVectorDimension)

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

构造器详细资料

JCudax

public JCudax()

方法详细资料

initialize

public static void initialize()

matrixSoftMaxPd

public static void matrixSoftMaxPd(double[] qkt,
                                   double[] errorMatrix,
                                   double[] grMatrix,
                                   int x,
                                   int y,
                                   double wordVectorDimension)

matrixSoftMaxPdFp32

public static void matrixSoftMaxPdFp32(float[] qkt,
                                       float[] errorMatrix,
                                       float[] grMatrix,
                                       int x,
                                       int y,
                                       float wordVectorDimension)