Loading [Contrib]/a11y/accessibility-menu.js
\u200E
文档[ torch 参数更多 ]torch.amin Edit on Github
[ torch 参数更多 ]torch.amin
torch.amin(input, dim, keepdim=False, *, out=None)
paddle.amin(x, axis=None, keepdim=False, name=None)
PyTorch 相比 Paddle 支持更多其他参数,具体如下:
参数映射
PyTorch |
PaddlePaddle |
备注 |
input |
x |
输入 Tensor,仅参数名不一致。 |
dim |
axis |
求最小值运算的维度,仅参数名不一致。 |
keepdim |
keepdim |
是否在输出 Tensor 中保留减小的维度。 |
out |
- |
输出 Tensor,Paddle 无此参数,需要转写。 |
转写示例
out: 指定输出
torch.amin(a, dim=0,out=y)
paddle.assign(paddle.amin(a, dim=0), y)
文档反馈