torch.prod¶
- torch.prod(input, *, dtype=None) Tensor ¶
返回
input
张量中所有元素的乘积。- 参数
input (Tensor) – 输入张量。
- 关键字参数
dtype (
torch.dtype
,可选) – 返回张量的所需数据类型。如果指定,则在执行操作之前将输入张量转换为dtype
。这有助于防止数据类型溢出。默认值:None。
示例
>>> 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。- 参数
- 关键字参数
dtype (
torch.dtype
,可选) – 返回张量的所需数据类型。如果指定,则在执行操作之前将输入张量转换为dtype
。这有助于防止数据类型溢出。默认值:None。
示例
>>> 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])