提交 1c1e9b00 编写于 作者: M Megvii Engine Team

fix(rng): init layout strides

GitOrigin-RevId: 9833d866dac7d0ed440ffebe55e19dedfb827db9
上级 9527859c
......@@ -334,6 +334,7 @@ struct _InferLayout<true> {
for (size_t i = 0; i < target_ndim; ++i) {
out_layout.shape[i] = ptr[i];
}
out_layout.init_contiguous_stride();
return out_layout;
}
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册