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

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

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

Nd4j.getWorkspaceManager介绍

[英]This method returns WorkspaceManager implementation to be used within this JVM process
[中]此方法返回要在此JVM进程中使用的WorkspaceManager实现

代码示例

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

/**
 * Assert that the specified workspace is open and active
 *
 * @param ws       Name of the workspace to assert open and active
 * @param errorMsg Message to include in the exception, if required
 */
public static void assertOpenAndActive(@NonNull String ws, @NonNull String errorMsg) throws ND4JWorkspaceException {
  if (!Nd4j.getWorkspaceManager().checkIfWorkspaceExistsAndActive(ws)) {
    throw new ND4JWorkspaceException(errorMsg);
  }
}

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

private void enforceExistsAndActive(@NonNull T arrayType){
    validateConfig(arrayType);
    if(scopeOutOfWs.contains(arrayType)){
      return;
    }

    if(!Nd4j.getWorkspaceManager().checkIfWorkspaceExistsAndActive(workspaceNames.get(arrayType))){
      throw new ND4JWorkspaceException("Workspace \"" + workspaceNames.get(arrayType) + "\" for array type " + arrayType
          + " is not open");
    }
  }
}

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

private static List<String> allOpenWorkspaces(){
  List<MemoryWorkspace> l = Nd4j.getWorkspaceManager().getAllWorkspacesForCurrentThread();
  List<String> workspaces = new ArrayList<>(l.size());
  for( MemoryWorkspace ws : l){
    if(ws.isScopeActive()) {
      workspaces.add(ws.getId());
    }
  }
  return workspaces;
}

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

private static List<String> allOpenWorkspaces(){
    List<MemoryWorkspace> l = Nd4j.getWorkspaceManager().getAllWorkspacesForCurrentThread();
    List<String> workspaces = new ArrayList<>(l.size());
    for( MemoryWorkspace ws : l){
      if(ws.isScopeActive()) {
        workspaces.add(ws.getId());
      }
    }
    return workspaces;
  }
}

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

@Override
public boolean isWorkspaceOpen(@NonNull T arrayType) {
  validateConfig(arrayType);
  if(!scopeOutOfWs.contains(arrayType)) {
    return Nd4j.getWorkspaceManager().checkIfWorkspaceExistsAndActive(getWorkspaceName(arrayType));
  }
  return true;
}

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

/**
 * Assert that no workspaces are currently open
 *
 * @param msg Message to include in the exception, if required
 */
public static void assertNoWorkspacesOpen(String msg) throws ND4JWorkspaceException {
  if (Nd4j.getWorkspaceManager().anyWorkspaceActiveForCurrentThread()) {
    List<MemoryWorkspace> l = Nd4j.getWorkspaceManager().getAllWorkspacesForCurrentThread();
    List<String> workspaces = new ArrayList<>(l.size());
    for (MemoryWorkspace ws : l) {
      if(ws.isScopeActive()) {
        workspaces.add(ws.getId());
      }
    }
    throw new ND4JWorkspaceException(msg + " - Open/active workspaces: " + workspaces);
  }
}

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

@Override
public MemoryWorkspace notifyScopeBorrowed(@NonNull T arrayType) {
  validateConfig(arrayType);
  enforceExistsAndActive(arrayType);
  if(scopeOutOfWs.contains(arrayType)){
    return Nd4j.getWorkspaceManager().scopeOutOfWorkspaces();
  } else {
    MemoryWorkspace ws = Nd4j.getWorkspaceManager().getWorkspaceForCurrentThread(
        getConfiguration(arrayType), getWorkspaceName(arrayType));
    return ws.notifyScopeBorrowed();
  }
}

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

@Override
public MemoryWorkspace notifyScopeEntered(@NonNull T arrayType) {
  validateConfig(arrayType);
  if(isScopedOut(arrayType)){
    return Nd4j.getWorkspaceManager().scopeOutOfWorkspaces();
  } else {
    MemoryWorkspace ws = Nd4j.getWorkspaceManager().getWorkspaceForCurrentThread(
        getConfiguration(arrayType), getWorkspaceName(arrayType));
    return ws.notifyScopeEntered();
  }
}

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

/**
 * Assert that the specified workspace is open, active, and is the current workspace
 *
 * @param ws       Name of the workspace to assert open/active/current
 * @param errorMsg Message to include in the exception, if required
 */
public static void assertOpenActiveAndCurrent(@NonNull String ws, @NonNull String errorMsg) throws ND4JWorkspaceException {
  if (!Nd4j.getWorkspaceManager().checkIfWorkspaceExistsAndActive(ws)) {
    throw new ND4JWorkspaceException(errorMsg + " - workspace is not open and active");
  }
  MemoryWorkspace currWs = Nd4j.getMemoryManager().getCurrentWorkspace();
  if (currWs == null || !ws.equals(currWs.getId())) {
    throw new ND4JWorkspaceException(errorMsg + " - not the current workspace (current workspace: "
        + (currWs == null ? null : currWs.getId()));
  }
}

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

private void initWorkspace() {
  workspace = Nd4j.getWorkspaceManager().createNewWorkspace(
      WorkspaceConfiguration.builder()
          .initialSize(memoryForGraph())
          .policyAllocation(AllocationPolicy.OVERALLOCATE)
          .policyLearning(LearningPolicy.FIRST_LOOP)
          .build());
  Nd4j.getWorkspaceManager().setWorkspaceForCurrentThread(workspace);
}

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

/**
 * This method detaches INDArray from current Workspace, and attaches it to Workspace with a given Id, if a workspace
 * with the given ID is open and active.
 *
 * If the workspace does not exist, or is not active, the array is detached from any workspaces.
 *
 * @param id ID of the workspace to leverage to
 * @return The INDArray, leveraged to the specified workspace (if it exists and is active) otherwise the detached array
 * @see #leverageTo(String)
 */
public INDArray leverageOrDetach(String id){
  if(!isAttached()){
    return this;
  }
  if(!Nd4j.getWorkspaceManager().checkIfWorkspaceExistsAndActive(id)){
    return detach();
  }
  return leverageTo(id);
}

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

this.id = workspaceId;
this.threadId = Thread.currentThread().getId();
this.guid = Nd4j.getWorkspaceManager().getUUID();
this.memoryManager = Nd4j.getMemoryManager();
this.deviceId = Nd4j.getAffinityManager().getDeviceForCurrentThread();

代码示例来源：origin: deeplearning4j/dl4j-examples

try(MemoryWorkspace ws = Nd4j.getWorkspaceManager().getAndActivateWorkspace(initialConfig, "SOME_ID")) {
  try (MemoryWorkspace ws = Nd4j.getWorkspaceManager().getAndActivateWorkspace(learningConfig, "OTHER_ID")) {
    INDArray array = Nd4j.create(100);
try(MemoryWorkspace ws1 = Nd4j.getWorkspaceManager().getAndActivateWorkspace(initialConfig, "SOME_ID")) {
  INDArray array = Nd4j.create(10, 10).assign(1.0f);
  INDArray sumRes;
  try(MemoryWorkspace ws2 = Nd4j.getWorkspaceManager().getAndActivateWorkspace(initialConfig, "THIRD_ID")) {
try(MemoryWorkspace ws1 = Nd4j.getWorkspaceManager().getAndActivateWorkspace(initialConfig, "SOME_ID")) {
  INDArray array1 = Nd4j.create(10, 10).assign(1.0f);
  INDArray array2;
  try(MemoryWorkspace ws = Nd4j.getWorkspaceManager().scopeOutOfWorkspaces()) {
  try (MemoryWorkspace ws1 = Nd4j.getWorkspaceManager().getAndActivateWorkspace(circularConfig, "CIRCULAR_ID")) {
    INDArray array = Nd4j.create(100);

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

@Override
public INDArray getActivation(INDArray in, boolean training) {
  if (training) {
    try(MemoryWorkspace ws = Nd4j.getWorkspaceManager().scopeOutOfWorkspaces()) {
      this.alpha = Nd4j.rand(in.shape(), l, u, Nd4j.getRandom());
    }
    INDArray inTimesAlpha = in.mul(alpha);
    BooleanIndexing.replaceWhere(in, inTimesAlpha, Conditions.lessThan(0));
  } else {
    this.alpha = null;
    double a = 0.5 * (l + u);
    return Nd4j.getExecutioner().execAndReturn(new RectifedLinear(in, a));
  }
  return in;
}

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

return this;
if (!Nd4j.getWorkspaceManager().checkIfWorkspaceExists(id)) {
  if(enforceExistence){
    throw new Nd4jNoSuchWorkspaceException(id);
MemoryWorkspace target = Nd4j.getWorkspaceManager().getWorkspaceForCurrentThread(id);

代码示例来源：origin: org.nd4j/nd4j-cuda-10.0

@Override
public MemoryWorkspace getWorkspaceForCurrentThread(@NonNull WorkspaceConfiguration configuration, @NonNull String id) {
  ensureThreadExistense();
  MemoryWorkspace workspace = backingMap.get().get(id);
  if (workspace == null) {
    workspace = newWorkspace(configuration, id);
    backingMap.get().put(id, workspace);
    if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING)
      pickReference(workspace);
  }
  return workspace;
}

代码示例来源：origin: org.nd4j/nd4j-cuda-10.0

@Override
public MemoryWorkspace createNewWorkspace(WorkspaceConfiguration configuration, String id) {
  ensureThreadExistense();
  MemoryWorkspace workspace = newWorkspace(configuration, id);
  backingMap.get().put(id, workspace);
  if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING)
    pickReference(workspace);
  return workspace;
}

代码示例来源：origin: org.nd4j/nd4j-cuda-10.0

@Override
public MemoryWorkspace createNewWorkspace(WorkspaceConfiguration configuration, String id, Integer deviceId) {
  ensureThreadExistense();
  MemoryWorkspace workspace = newWorkspace(configuration, id, deviceId);
  backingMap.get().put(id, workspace);
  if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING)
    pickReference(workspace);
  return workspace;
}

代码示例来源：origin: org.nd4j/nd4j-cuda-10.0

@Override
public MemoryWorkspace createNewWorkspace(@NonNull WorkspaceConfiguration configuration) {
  ensureThreadExistense();
  MemoryWorkspace workspace = newWorkspace(configuration);
  backingMap.get().put(workspace.getId(), workspace);
  if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING)
    pickReference(workspace);
  return workspace;
}

代码示例来源：origin: org.nd4j/nd4j-cuda-10.0

@Override
public MemoryWorkspace createNewWorkspace() {
  ensureThreadExistense();
  MemoryWorkspace workspace = newWorkspace(defaultConfiguration);
  backingMap.get().put(workspace.getId(), workspace);
  if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING)
    pickReference(workspace);
  return workspace;
}

内容来源于网络，如有侵权，请联系作者删除！

Nd4j

相关文章

    查看更多

    热门标签

    更多
    JavaquerypythonNode开发语言requestUtil数据库Table后端算法LoggerMessageElementParser

    最新文章

    更多

    Nd4j类方法

    • 蜀ICP备13028337号-1 大数据知识库 https://www.saoniuhuo.com © All rights reserved
    • 本站内容来源互联网,如果侵犯您的权益请联系我们删除, 联系方式：448109455@qq.com