pytorch tensor創建

create tensor 1. torch.tensor():從data創建tensor data可以是list,numpy;dtype默認與data一致。 2. torch.from_numpy(ndarray):從numpy創建tensor 這裏創建的tensor與原ndarray共享內存,一個改變另一個就會改變。 3. torch.zeros():依size創建全0張量   torch.z
相關文章
相關標籤/搜索