torch.full()简要介绍

两分先生

于 2022-01-16 20:48:52 发布

阅读量1.1k

点赞数

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

本文链接：https://blog.csdn.net/weixin_45335726/article/details/122528546

版权

pytorch 专栏收录该内容

9 篇文章

订阅专栏

定义

torch.full(size,num)

用法

话不多说，直接看示例

示例

import torch

a = torch.full((4,4),3)
b = torch.full((5,6),9)
print(a)
print(b)

>>>tensor([[3, 3, 3, 3],
           [3, 3, 3, 3],
           [3, 3, 3, 3],
           [3, 3, 3, 3]])
tensor([[9, 9, 9, 9, 9, 9],
        [9, 9, 9, 9, 9, 9],
        [9, 9, 9, 9, 9, 9],
        [9, 9, 9, 9, 9, 9],
        [9, 9, 9, 9, 9, 9]])