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)
还没有任何评论哟~
