torch.ones_like()

Stig_Q

于 2022-11-29 16:14:49 发布

阅读量1.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：机器学习 ML 文章标签：深度学习 python pytorch

本文链接：https://blog.csdn.net/qq_37006625/article/details/128100125

机器学习 ML 专栏收录该内容

19 篇文章

订阅专栏

torch.ones_like(input)是PyTorch中的一个函数，用于创建一个与输入张量input形状相同的张量，并将其所有元素初始化为1。此函数在需要快速填充1的张量时非常实用。例如，当您需要一个与现有张量大小匹配的全1矩阵作为初始值或占位符时，可以使用这个函数。在提供的示例中，当input是一个空的(2,3)张量时，torch.ones_like(input)将返回一个2x3的全1张量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

TORCH.ONES_LIKE

torch.ones_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format) → Tensor

返回一个填充了标量值1的张量，shape 与 input 相同

torch.ones_like(input)相当于torch.ones(input)

例子

>>> input = torch.empty(2, 3)
>>> torch.ones_like(input)

tensor([[ 1.,  1.,  1.],
        [ 1.,  1.,  1.]])