concat static method

Tensor concat(
  1. List<Tensor> tensors
)

Implementation

static Tensor concat(List<Tensor> tensors) {
  final ptr = calloc<ffi.Pointer<ffi.Void>>(tensors.length);
  for (int i = 0; i < tensors.length; i++) {
    ptr[i] = tensors[i]._handle;
  }
  final h = engine.concatTensors(ptr, tensors.length);
  calloc.free(ptr);
  return Tensor._raw(h, [
    tensors[0].shape[0],
    tensors[0].shape[1] * tensors.length,
  ]);
}