conv2d¶
dragon.vm.torch.nn.functional.conv2d(
input,
weight,
bias=None,
stride=1,
padding=0,
dilation=1,
groups=1
)[source]¶Apply the 2d convolution to input.
- Parameters:
- input (dragon.vm.torch.Tensor) – The input tensor.
- weight (dragon.vm.torch.Tensor) – The weight tensor.
- bias (dragon.vm.torch.Tensor, optional) – The bias tensor.
- stride (Union[int, Sequence[int]], optional, default=1) – The stride of convolution window.
- padding (Union[int, Sequence[int]], optional, default=0) – The zero padding size.
- dilation (Union[int, Sequence[int]], optional, default=1) – The rate of dilated kernel.
- groups (int, optional, default=1) – The number of groups to split input channels.
- Returns:
dragon.vm.torch.Tensor – The output tensor.
See also