本文整理匯總了Java中org.nd4j.linalg.factory.Nd4j.ones方法的典型用法代碼示例。如果您正苦於以下問題:Java Nd4j.ones方法的具體用法?Java Nd4j.ones怎麽用?Java Nd4j.ones使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類org.nd4j.linalg.factory.Nd4j
的用法示例。
在下文中一共展示了Nd4j.ones方法的2個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Java代碼示例。
示例1: NDMatrix
import org.nd4j.linalg.factory.Nd4j; //導入方法依賴的package包/類
/**
* Constructor. Fill the matrix with given value.
*/
public NDMatrix(int rows, int cols, double value) {
if (value == 0.0)
A = Nd4j.zeros(rows, cols);
else if (value == 1.0)
A = Nd4j.ones(rows, cols);
else
A = Nd4j.zeros(rows, cols).addi(value);
}
示例2: ones
import org.nd4j.linalg.factory.Nd4j; //導入方法依賴的package包/類
/** @return {@link INDArray} */
public INDArray ones () {
return Nd4j.ones(1, layerSize);
}