29 lines
1.0 KiB
ReStructuredText
29 lines
1.0 KiB
ReStructuredText
mindspore.mint.nn.ConstantPad2d
|
||
===============================
|
||
|
||
.. py:class:: mindspore.mint.nn.ConstantPad2d(padding, value)
|
||
|
||
根据参数 `padding` 以及 `value` 对输入 `input` 最后2维进行填充。
|
||
|
||
更多参考详见 :func:`mindspore.mint.nn.functional.pad`。
|
||
|
||
.. warning::
|
||
这是一个实验性API,后续可能修改或删除。
|
||
|
||
参数:
|
||
- **padding** (Union[int, tuple, list]) - 指定填充的大小。
|
||
- **value** (Union[int, float]) - 指定填充值。
|
||
|
||
输入:
|
||
- **input** (Tensor) - 输入Tensor,shape为 :math:`(N, *)`,其中 :math:`*` 表示任意维度。
|
||
|
||
输出:
|
||
Tensor,填充后的Tensor。
|
||
|
||
异常:
|
||
- **TypeError** - `padding` 不是一个integer或包含4个int的tuple或者list。
|
||
- **TypeError** - `input` 不是Tensor。
|
||
- **TypeError** - `value` 既不是int,也不是float。
|
||
- **ValueError** - `padding` 含有负数。
|
||
- **ValueError** - `padding` 是tuple或list,且长度和Tensor的维度不匹配。
|