Adding the [] accessor

assaydepot · Mar 14, 2024 · c9ad1f3 · c9ad1f3
1 parent f1585a8
commit c9ad1f3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ext/candle/src/lib.rs b/ext/candle/src/lib.rs
@@ -46,6 +46,7 @@ fn init(ruby: &Ruby) -> RbResult<()> {
     rb_tensor.define_method("squeeze", method!(RbTensor::squeeze, 1))?;
     rb_tensor.define_method("unsqueeze", method!(RbTensor::unsqueeze, 1))?;
     rb_tensor.define_method("get", method!(RbTensor::get, 1))?;
+    rb_tensor.define_method("[]", method!(RbTensor::get, 1))?;
     rb_tensor.define_method("transpose", method!(RbTensor::transpose, 2))?;
     rb_tensor.define_method("narrow", method!(RbTensor::narrow, 3))?;
     rb_tensor.define_method("argmax_keepdim", method!(RbTensor::argmax_keepdim, 1))?;