PythonNumpy:ascontiguousarray用例?

2024-06-06 13:13:31 发布

您现在位置:Python中文网/ 问答频道 /正文

我知道ascontiguousarray在内存中返回一个连续数组,但是有人能告诉我什么时候真的需要它吗?i、 e.没有此功能,结果是否错误?或者这会使后续操作更有效

我写了一个卷积函数,我用as_strided来改变它的形状和步幅,然后einsum与内核相乘,最后是moveaxis,所以最后的形状是(N,C,H,W),就像pytorch一样。那么在这种情况下,您认为我应该在输出之前使用ascontiguousarray吗?谢谢


Tags: 函数内存功能as错误pytorch数组内核