torch笔记1

人走茶凉 提交于 2020-01-14 19:43:47

tensor

1、torch.(…)和torch.([…])的区别

前者参数是tensor的维度,后者参数是tensor中的数据

2、 size和shape的区别

shape is an attribute,while size is a function.shape用法
size用法
输出都是同一个tensor

3、 在函数后加_后缀表示对tensor进行就地操作(ex:X.t_()表示对X转置)

4、torch的tensor类

torch.tensor(data,dtype=None,device=None,requires_grad=None)

data - 可以是list, tuple, numpy array, scalar或其他类型,如下图分别是错误和正确的data
在这里插入图片描述
dtype - 可以返回想要的tensor类型
在这里插入图片描述
device - 可以指定返回的设备,“cuda"或"cpu”
requires_grad -可以指定是否进行记录图的操作,默认为False

5、rand()和randn()区别

在这里插入图片描述
rand()生成tensor内的数据是符合[0,1)范围内的均匀分布的;
randn()生成tensor内的数据是符合标准正态分布(standard normal distribution)的。

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!