checkShape() — pytorch Function Reference

Architecture documentation for the checkShape() function in Tensor.java from the pytorch codebase.

Function java AndroidRuntime JavaAPI calls 1 called by 4

Entity Profile

AndroidRuntime→ JavaAPI→ checkShape() — pytorch Function Reference

Dependency Diagram

graph TD
  384ee7bc_0c17_aa62_2773_4d58b6ffdfc5["checkShape()"]
  a3810b96_9142_0fab_7722_abfe805aec76["fromBlobUnsigned()"]
  a3810b96_9142_0fab_7722_abfe805aec76 -->|calls| 384ee7bc_0c17_aa62_2773_4d58b6ffdfc5
  7d987baf_1d44_31d6_f128_18964ed36ba8["fromBlob()"]
  7d987baf_1d44_31d6_f128_18964ed36ba8 -->|calls| 384ee7bc_0c17_aa62_2773_4d58b6ffdfc5
  ebadc71e_cab5_792b_ebff_10cde5eff184["Tensor()"]
  ebadc71e_cab5_792b_ebff_10cde5eff184 -->|calls| 384ee7bc_0c17_aa62_2773_4d58b6ffdfc5
  0904339d_a9fa_c19b_1171_129c4c841cd6["numel()"]
  0904339d_a9fa_c19b_1171_129c4c841cd6 -->|calls| 384ee7bc_0c17_aa62_2773_4d58b6ffdfc5
  a8703cc9_ca82_8058_0c67_94bdd5d5f63f["checkArgument()"]
  384ee7bc_0c17_aa62_2773_4d58b6ffdfc5 -->|calls| a8703cc9_ca82_8058_0c67_94bdd5d5f63f
  style 384ee7bc_0c17_aa62_2773_4d58b6ffdfc5 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/src/main/java/org/pytorch/Tensor.java lines 686–691

  private static void checkShape(long[] shape) {
    checkArgument(shape != null, ERROR_MSG_SHAPE_NOT_NULL);
    for (int i = 0; i < shape.length; i++) {
      checkArgument(shape[i] >= 0, ERROR_MSG_SHAPE_NON_NEGATIVE);
    }
  }