testEqTensor() — pytorch Function Reference

Architecture documentation for the testEqTensor() function in PytorchTestBase.java from the pytorch codebase.

Function java AndroidRuntime TensorAPI calls 1

Entity Profile

AndroidRuntime→ TensorAPI→ testEqTensor() — pytorch Function Reference

Dependency Diagram

graph TD
  500d1a47_0f5a_9b79_d08c_e88bcfdea1ff["testEqTensor()"]
  fd3711c0_c2a1_0848_309d_ba2ef1be4466["loadModel()"]
  500d1a47_0f5a_9b79_d08c_e88bcfdea1ff -->|calls| fd3711c0_c2a1_0848_309d_ba2ef1be4466
  style 500d1a47_0f5a_9b79_d08c_e88bcfdea1ff fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/src/androidTest/java/org/pytorch/PytorchTestBase.java lines 80–103

  @Test
  public void testEqTensor() throws IOException {
    final long[] inputTensorShape = new long[] {1, 3, 224, 224};
    final long numElements = Tensor.numel(inputTensorShape);
    final float[] inputTensorData = new float[(int) numElements];
    for (int i = 0; i < numElements; ++i) {
      inputTensorData[i] = i;
    }
    final Tensor inputTensor = Tensor.fromBlob(inputTensorData, inputTensorShape);

    final Module module = loadModel(TEST_MODULE_ASSET_NAME);
    final IValue input = IValue.from(inputTensor);
    assertTrue(input.isTensor());
    assertTrue(inputTensor == input.toTensor());
    final IValue output = module.runMethod("eqTensor", input);
    assertTrue(output.isTensor());
    final Tensor outputTensor = output.toTensor();
    assertNotNull(outputTensor);
    assertArrayEquals(inputTensorShape, outputTensor.shape());
    float[] outputData = outputTensor.getDataAsFloatArray();
    for (int i = 0; i < numElements; i++) {
      assertTrue(inputTensorData[i] == outputData[i]);
    }
  }