torch.Conv3dShape

export type Conv3dShape<S extends Shape, OutChannels extends number> = S extends readonly [
  infer B extends number,
  number,
  number,
  number,
  number,
]
  ? readonly [B, OutChannels, number, number, number]
  : S extends readonly [number, number, number, number]
    ? readonly [OutChannels, number, number, number]
    : DynamicShape;

Sextends Shape

OutChannelsextends number

Output shape for 3D convolution. Input: [B, Cin, D, H, W], Output: [B, Cout, outD, outH, outW]