free method

void free()

Frees the allocated GPU Memory for both value and gradient.

Implementation

void free() {
  for (GPUTensor tensor in subMap.values) {
    tensor.free();
  }

  GPUEngine.free(id);

  if (localGrad == true) {
    GPUEngine.free('${id}_grad');
  }
}