weixin_47267930的博客

原创 PyTorch中一些常用的函数

1. torch.as_tensor(data, dtype=None,device=None)->Tensor: 为data生成tensor。这里data的数据类型可以是list、tuple、ndarray、scalar等若data已经是tensor，且dtype和device与参数相同，则生成的tensor会和data共享内存。如果data是ndarray, 且dtype对应，devices为cpu，则同样共享内存。其他情况不共享内存。import torchimport num.