torch.ones¶
- torch.ones(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) Tensor ¶
返回一个填充了标量值 1 的 tensor,其形状由可变参数
size
定义。- 参数
size (int...) – 定义输出 tensor 形状的整数序列。可以是可变数量的参数,也可以是列表或元组等集合。
- 关键字参数
out (Tensor, optional) – 输出 tensor。
dtype (
torch.dtype
, optional) – 返回 tensor 的期望数据类型。默认值:如果为None
,则使用全局默认设置(参见torch.set_default_dtype()
)。layout (
torch.layout
, optional) – 返回 Tensor 的期望布局。默认值:torch.strided
。device (
torch.device
, optional) – 返回 tensor 的期望设备。默认值:如果为None
,则使用默认 tensor 类型的当前设备(参见torch.set_default_device()
)。对于 CPU tensor 类型,device
将是 CPU;对于 CUDA tensor 类型,将是当前的 CUDA 设备。requires_grad (bool, optional) – 如果 autograd 应该记录返回 tensor 上的操作。默认值:
False
。
示例
>>> torch.ones(2, 3) tensor([[ 1., 1., 1.], [ 1., 1., 1.]]) >>> torch.ones(5) tensor([ 1., 1., 1., 1., 1.])