GPUTensor<T>.tensor3D constructor

GPUTensor<T>.tensor3D(
  1. Tensor3D initialValue, {
  2. GPUNode? creator,
  3. String id = "EMPTY",
  4. bool? requiresGrad,
})

Implementation

GPUTensor.tensor3D(Tensor3D initialValue, {this.creator, this.id = "EMPTY", bool? requiresGrad}) {
  localGrad = requiresGrad ?? globalGrad;
  if (id == "EMPTY") {
    id = _generateId();
  }
  int depth = initialValue.length;
  int height = depth > 0 ? initialValue[0].length : 0;
  int width = height > 0 ? initialValue[0][0].length : 0;
  shape = [depth, height, width];
  _allocateEmptyInVram();
  _pushInitialValue(initialValue);
}