fix incorrect nidex type

apache · Aug 21, 2019 · 3a1d25a · 3a1d25a
1 parent b9f9cac
commit 3a1d25a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/3rdparty/mshadow/mshadow/tensor.h b/3rdparty/mshadow/mshadow/tensor.h
@@ -506,7 +506,7 @@ struct Tensor: public TRValue<Tensor<Device, dimension, DType>,
    * \param idx index
    * \return the result tensor
    */
-  MSHADOW_XINLINE Tensor<Device, kSubdim, DType> operator[](int idx) const {
+  MSHADOW_XINLINE Tensor<Device, kSubdim, DType> operator[](index_t idx) const {
     return Tensor<Device, kSubdim, DType>(dptr_ + this->MemSize<1>() * idx,
                                           shape_.SubShape(), stride_, stream_);
   }