Skip to contents

3D Convolution

Source: R/api.R
nv_conv3d.Rd

Torch-style 3D convolution in NCDHW layout. x is [batch, in_channels, depth, height, width], weight is [out_channels, in_channels / groups, kD, kH, kW]. Asymmetric padding (e.g. causal temporal padding) is available via prim_convolution().

Usage

nv_conv3d(
  x,
  weight,
  stride = 1L,
  padding = 0L,
  dilation = 1L,
  groups = 1L,
  precision = "highest"
)

Arguments

x

(arrayish)
[N, C_in, H, W].

weight

(arrayish)
[C_out, C_in / groups, kH, kW].

stride, padding, dilation

(integer())
Length 1 or 3.

groups

(integer(1))
Grouped/depthwise convolution.

precision

(character(1))
"highest", "high" or "default".

Value

arrayish [N, C_out, out_D, out_H, out_W].

See also

nv_conv1d(), nv_conv2d(), prim_convolution().