python——Contiguous用法

x.is_contiguous()——判断tensor是否连续

x.contiguous()——把tensor变成在内存中连续分布的形式

需要变成连续分布的情况:

contiguous:view只能用在contiguous的variable上。如果在view之前用了transpose, permute等,需要用contiguous()来返回一个contiguous copy。

参考资料:https://blog.csdn.net/appleml/article/details/80143212

import torch
x = torch.ones(10, 10)
print(x.is_contiguous())
print(x.transpose(0, 1).is_contiguous())
print(x.transpose(0, 1).contiguous().is_contiguous()) 

结果:

True
False
True

 

你可能感兴趣的:(python——Contiguous用法)