... ie, in a 3d vector, it would be the least contiguous dimension, Z, or pages, 3rd dim etc. ... PyTorch, which supports arrays allocated on the GPU.
確定! 回上一頁