本文简要介绍python语言中 torch.prod
的用法。
用法:
torch.prod(input, *, dtype=None) → Tensor
input(Tensor) -输入张量。
dtype(
torch.dtype
, 可选的) -返回张量的所需数据类型。如果指定,则在执行操作之前将输入张量强制转换为dtype
。这对于防止数据类型溢出很有用。默认值:无。dtype(
torch.dtype
, 可选的) -返回张量的所需数据类型。如果指定,则在执行操作之前将输入张量强制转换为dtype
。这对于防止数据类型溢出很有用。默认值:无。返回
input
张量中所有元素的乘积。例子:
>>> a = torch.randn(1, 3) >>> a tensor([[-0.8020, 0.5428, -1.5854]]) >>> torch.prod(a) tensor(0.6902)
torch.prod(input, dim, keepdim=False, *, dtype=None) → Tensor
返回给定维度
dim
中input
张量的每一行的乘积。如果
keepdim
是True
,则输出张量的大小与input
相同,但在维度dim
中它的大小为 1。否则,dim
被挤压(参见torch.squeeze()
),结果在比input
少 1 个维度的输出张量中。例子:
>>> a = torch.randn(4, 2) >>> a tensor([[ 0.5261, -0.3837], [ 1.1857, -0.2498], [-1.1646, 0.0705], [ 1.1131, -1.0629]]) >>> torch.prod(a, 1) tensor([-0.2018, -0.2962, -0.0821, -1.1831])
参数:
关键字参数:
参数:
关键字参数:
相关用法
- Python PyTorch promote_types用法及代码示例
- Python PyTorch profile用法及代码示例
- Python PyTorch prof用法及代码示例
- Python PyTorch prepare用法及代码示例
- Python PyTorch positive用法及代码示例
- Python PyTorch powerSGD_hook用法及代码示例
- Python PyTorch pca_lowrank用法及代码示例
- Python PyTorch pixel_shuffle用法及代码示例
- Python PyTorch pinv用法及代码示例
- Python PyTorch put_metric用法及代码示例
- Python PyTorch pad_sequence用法及代码示例
- Python PyTorch pow用法及代码示例
- Python PyTorch phase_vocoder用法及代码示例
- Python PyTorch pop用法及代码示例
- Python PyTorch polar用法及代码示例
- Python PyTorch poisson用法及代码示例
- Python PyTorch pack_sequence用法及代码示例
- Python PyTorch pad用法及代码示例
- Python PyTorch pad_packed_sequence用法及代码示例
- Python PyTorch pixel_unshuffle用法及代码示例
- Python PyTorch polygamma用法及代码示例
- Python PyTorch permute用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.prod。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。