adds error checking for tf.Buffer.get(outOfRangeLocation)

src/buffer_test.ts

@@ -34,6 +34,16 @@ describeWithFlags('tf.buffer', ALL_ENVS, () => {
     expectArraysClose(buff.values, new Float32Array([1.3, 0, 0, 2.9, 0, 0]));
   });
 
+  it('get() out of range throws', () => {
+    const t = tf.tensor([1, 2, 3, 4, 5, 6, 7, 8], [2, 2, 2]);
+
+    const buff = t.bufferSync();
+    expect(buff.get(0, 0, 0)).toBeCloseTo(1);
+    expect(buff.get(0, 0, 1)).toBeCloseTo(2);
+    expect(() => buff.get(0, 0, 2))
+        .toThrowError(/Requested out of range element/);
+  });
+
   it('int32', () => {
     const buff = tf.buffer([2, 3], 'int32');
     buff.set(1.3, 0, 0);

src/tensor.ts

@@ -98,6 +98,13 @@ export class TensorBuffer<R extends Rank, D extends DataType = 'float32'> {
     if (locs.length === 0) {
       locs = [0];
     }
+    for (const i in locs) {
+      if (locs[i] < 0 || locs[i] >= this.shape[i]) {
+        const msg = `Requested out of range element at ${locs}. ` +
+            `  Buffer shape=${this.shape}`;
+        throw new Error(msg);
+      }
+    }
     let index = locs[locs.length - 1];
     for (let i = 0; i < locs.length - 1; ++i) {
       index += this.strides[i] * locs[i];