PyTorch是由Facebook开发的开源机器学习库。它用于深度神经网络和自然语言处理。
函数torch.ones()
返回一个由标量值1填充的张量,其形状由变量参数size定义。
用法:torch.ones(size, out=None)
参数:
size:定义输出张量形状的整数序列
out (Tensor, optional):输出张量
返回类型:用标量值1填充的张量,其形状与尺寸相同。
代码1:
# Importing the PyTorch library
import torch
# Applying the ones function and
# storing the resulting tensor in 't'
a = torch.ones([3, 4])
print("a = ", a)
b = torch.ones([1, 5])
print("b = ", b)
c = torch.ones([5, 1])
print("c = ", c)
d = torch.ones([3, 3, 2])
print("d = ", d)
输出:
a = tensor([[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]]) b = tensor([[1., 1., 1., 1., 1.]]) c = tensor([[1.], [1.], [1.], [1.], [1.]]) d = tensor([[[1., 1.], [1., 1.], [1., 1.]], [[1., 1.], [1., 1.], [1., 1.]], [[1., 1.], [1., 1.], [1., 1.]]])
相关用法
- Python PyTorch sin()用法及代码示例
- Python PyTorch sinh()用法及代码示例
- Python PyTorch cosh()用法及代码示例
- Python PyTorch tanh()用法及代码示例
- Python PyTorch cos()用法及代码示例
- Python PyTorch tan()用法及代码示例
- Python PyTorch asin()用法及代码示例
- Python PyTorch acos()用法及代码示例
- Python PyTorch atan()用法及代码示例
- Python PyTorch zeros()用法及代码示例
- Python Pytorch arange()用法及代码示例
- Python Pytorch linspace()用法及代码示例
- Python Pytorch range()用法及代码示例
- Python Pytorch logspace()用法及代码示例
- Python Pytorch eye()用法及代码示例
- Python Pytorch empty()用法及代码示例
- Python Pytorch full()用法及代码示例
- Python PyTorch floor()用法及代码示例
- Python PyTorch fmod()用法及代码示例
- Python PyTorch log()用法及代码示例
注:本文由纯净天空筛选整理自sanskar27jain大神的英文原创作品 Python Pytorch ones() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。