当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python Pytorch full()用法及代码示例


PyTorch是由Facebook开发的开源机器学习库。它用于深度神经网络和自然语言处理。

函数torch.full()返回一个大小为fill_value的张量的张量。

用法:torch.ones(size, fill_value, out=None)

参数
size:定义输出张量形状的整数序列
fill_value:用于填充输出张量的数字。
out (Tensor, optional):输出张量

返回类型:张量



代码1:

# Importing the PyTorch library 
import torch 
  
  
# Applying the full function and 
# storing the resulting tensor in 'a' 
a = torch.full([3, 4], 3) 
print("a = ", a) 
  
b = torch.full([2, 5], 3.5) 
print("b = ", b)

输出:

a =  tensor([[3., 3., 3., 3.],
        [3., 3., 3., 3.],
        [3., 3., 3., 3.]])
b =  tensor([[3.5000, 3.5000, 3.5000, 3.5000, 3.5000],
        [3.5000, 3.5000, 3.5000, 3.5000, 3.5000]])

相关用法


注:本文由纯净天空筛选整理自sanskar27jain大神的英文原创作品 Python Pytorch full() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。