checkShape() — pytorch Function Reference

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

Function java AndroidRuntime TensorAPI calls 1 called by 4

Entity Profile

AndroidRuntime→ TensorAPI→ checkShape() — pytorch Function Reference

Dependency Diagram

graph TD
  3326fa3f_4fcb_175c_bb45_ea4994768ac1["checkShape()"]
  6be65676_2065_aa9b_3e47_fa15f51dd3fe["fromBlobUnsigned()"]
  6be65676_2065_aa9b_3e47_fa15f51dd3fe -->|calls| 3326fa3f_4fcb_175c_bb45_ea4994768ac1
  fff97aed_e384_1976_ed4b_d94a2ed220dd["fromBlob()"]
  fff97aed_e384_1976_ed4b_d94a2ed220dd -->|calls| 3326fa3f_4fcb_175c_bb45_ea4994768ac1
  84782774_5716_4468_9ff1_c032204897d3["Tensor()"]
  84782774_5716_4468_9ff1_c032204897d3 -->|calls| 3326fa3f_4fcb_175c_bb45_ea4994768ac1
  40db4560_92cc_72c7_f2bb_938fd58a0403["numel()"]
  40db4560_92cc_72c7_f2bb_938fd58a0403 -->|calls| 3326fa3f_4fcb_175c_bb45_ea4994768ac1
  71889387_1e63_1ff3_2450_e3018d61b4ee["checkArgument()"]
  3326fa3f_4fcb_175c_bb45_ea4994768ac1 -->|calls| 71889387_1e63_1ff3_2450_e3018d61b4ee
  style 3326fa3f_4fcb_175c_bb45_ea4994768ac1 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);
    }
  }