org.nd4j.linalg.factory.Nd4j.createUninitializedDetached()方法的使用及代码示例

x33g5p2x  于2022-01-24 转载在 其他  
字(4.2k)|赞(0)|评价(0)|浏览(107)

本文整理了Java中org.nd4j.linalg.factory.Nd4j.createUninitializedDetached()方法的一些代码示例,展示了Nd4j.createUninitializedDetached()的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。Nd4j.createUninitializedDetached()方法的具体详情如下:
包路径:org.nd4j.linalg.factory.Nd4j
类名称:Nd4j
方法名:createUninitializedDetached

Nd4j.createUninitializedDetached介绍

[英]Cretes uninitialized INDArray detached from any (if any) workspace
[中]Cretes未初始化的INDArray从任何(如果有的话)工作区分离

代码示例

代码示例来源:origin: deeplearning4j/nd4j

  1. /**
  2. * Cretes uninitialized INDArray detached from any (if any) workspace
  3. *
  4. * @param shape
  5. * @return
  6. */
  7. public static INDArray createUninitializedDetached(int[] shape) {
  8. return createUninitializedDetached(shape, Nd4j.order());
  9. }

代码示例来源:origin: deeplearning4j/nd4j

  1. /**
  2. * Cretes uninitialized INDArray detached from any (if any) workspace
  3. *
  4. * @param shape
  5. * @return
  6. */
  7. public static INDArray createUninitializedDetached(long[] shape) {
  8. return createUninitializedDetached(shape, Nd4j.order());
  9. }

代码示例来源:origin: org.nd4j/nd4j-api

  1. /**
  2. * Cretes uninitialized INDArray detached from any (if any) workspace
  3. *
  4. * @param shape
  5. * @return
  6. */
  7. public static INDArray createUninitializedDetached(int[] shape) {
  8. checkShapeValues(shape);
  9. //ensure shapes that wind up being scalar end up with the write shape
  10. return createUninitializedDetached(shape, Nd4j.order());
  11. }

代码示例来源:origin: CampagneLaboratory/variationanalysis

  1. private void keepLongestMask(int minibatchSize, INDArray mask, int[] randomIndex1, int[] randomIndex2) {
  2. if (mask == null) return;
  3. INDArray[] tmpBuffer = new INDArray[minibatchSize];
  4. // Find the longest mask and keep it as mixup ask:
  5. for (int exampleIndex = 0; exampleIndex < minibatchSize; exampleIndex++) {
  6. int random1 = randomIndex1[exampleIndex];
  7. int random2 = randomIndex2[exampleIndex];
  8. final INDArray mask1 = mask.getRow(random1);
  9. final INDArray mask2 = mask.getRow(random2);
  10. tmpBuffer[exampleIndex] = Nd4j.createUninitializedDetached(mask1.shape());
  11. if (mask1.sub(mask2).sumNumber().doubleValue() < 0) {
  12. // mask2 has more 1s than mask1, use mask2:
  13. Nd4j.copy(mask2, tmpBuffer[exampleIndex]);
  14. } else {
  15. Nd4j.copy(mask1, tmpBuffer[exampleIndex]);
  16. }
  17. }
  18. for (int exampleIndex = 0; exampleIndex < minibatchSize; exampleIndex++) {
  19. // assign tmpBuffer[inputIndex] back into the minibatch:
  20. mask.putRow(exampleIndex, tmpBuffer[exampleIndex]);
  21. }
  22. }

代码示例来源:origin: CampagneLaboratory/variationanalysis

  1. inputs[sampleIndex][index] = Nd4j.createUninitializedDetached(inputShape, 'f');
  2. inputMasks[sampleIndex][index] = needMask ? Nd4j.createUninitializedDetached(
  3. domainDescriptor.getInputMaskShape(size, input), 'f'
  4. ) : null;
  5. for (int sampleIndex = 0; sampleIndex < sampleIndices.length; sampleIndex++) {
  6. labelMappers[index][sampleIndex] = domainDescriptor.getLabelMapper(label);
  7. labels[sampleIndex][index] = Nd4j.createUninitializedDetached(domainDescriptor.getLabelShape(size, label), 'f');
  8. labelMasks[sampleIndex][index] = needMask ? Nd4j.createUninitializedDetached(
  9. domainDescriptor.getLabelMaskShape(size, label), 'f'
  10. ) : null;

代码示例来源:origin: CampagneLaboratory/variationanalysis

  1. inputs[index] = Nd4j.createUninitializedDetached(domainDescriptor.getInputShape(size, input),'f');
  2. featureMappers[index] = domainDescriptor.getFeatureMapper(input);
  3. index += 1;
  4. labels[index] = Nd4j.createUninitializedDetached(domainDescriptor.getLabelShape(size, label),'f');
  5. labelMappers[index] = domainDescriptor.getLabelMapper(label);
  6. index++;

代码示例来源:origin: CampagneLaboratory/variationanalysis

  1. inputs[index] = Nd4j.createUninitializedDetached(inputShape, 'f');
  2. featureMappers[index] = domainDescriptor.getFeatureMapper(input);
  3. boolean needMask = featureMappers[index].hasMask();
  4. inputMasks[index] = needMask ? Nd4j.createUninitializedDetached(domainDescriptor.getInputMaskShape(size, input), 'f') : null;
  5. labels[index] = Nd4j.createUninitializedDetached(domainDescriptor.getLabelShape(size, label), 'f');
  6. labelMasks[index] = Nd4j.createUninitializedDetached(domainDescriptor.getLabelMaskShape(size, label), 'f');

代码示例来源:origin: org.deeplearning4j/deeplearning4j-cuda-9.2

  1. if (training) {
  2. if(meanCache == null || meanCache.length() < mean.length()){
  3. meanCache = Nd4j.createUninitializedDetached((int)mean.length());
  4. if(Nd4j.dataType() == DataBuffer.Type.HALF){
  5. try(MemoryWorkspace ws = Nd4j.getMemoryManager().scopeOutOfWorkspaces()) {
  6. varCache = Nd4j.createUninitializedDetached((int)mean.length());
  7. if(Nd4j.dataType() == DataBuffer.Type.HALF){
  8. try(MemoryWorkspace ws = Nd4j.getMemoryManager().scopeOutOfWorkspaces()) {

相关文章