Function testArrayCopy

web/tests/node/test_tensor.js:36–46

(dtype, arrayType)

// Test ndarray
function testArrayCopy(dtype, arrayType) {
  let data = [1, 2, 3, 4, 5, 6];
  let a = tvm.empty([2, 3], dtype).copyFrom(data);

  assert(a.device.toString() == "cpu:0");
  assert(a.shape[0] == 2 && a.shape[1] == 3);

  let ret = a.toArray();
  assert(ret instanceof arrayType);
  assert(ret.toString() == arrayType.from(data).toString());
}

test("array copy", () => {
  tvm.withNewScope(() => {