Advertisement

Dl4j-fit(DataSetIterator iterator)源代码第5章正向传播

阅读量:

随后进行的操作是执行preOutput()函数所包含的以下两个语句:

复制代码
    INDArray b = getParam(DefaultParamInitializer.BIAS_KEY);
    INDArray W = getParam(DefaultParamInitializer.WEIGHT_KEY);
    
      
      
    

用于提取当前层级的权重参数与偏置值。随后进行的操作包括对输入数据的有效性验证,以及判断是否启用dropoutConnect机制,但鉴于当前所采用的网络结构并未涉及该功能,此处暂不展开讨论。

复制代码
     //Input validation:
    if (input.rank() != 2 || input.columns() != W.rows()) {
    if (input.rank() != 2) {
        throw new DL4JInvalidInputException("Input that is not a matrix; expected matrix (rank 2), got rank "
                        + input.rank() + " array with shape " + Arrays.toString(i

全部评论 (0)

还没有任何评论哟~