Class Conv2d

Hierarchy

_ConvNd
- Conv2d

Index

Constructors

Properties

bias dilation groups in_channels kernel_size out_channels padding stride weight

Methods

forward parameters register

Constructors

constructor

new Conv2d(
    in_channels: number,
    out_channels: number,
    kernel_size: number | number[],
    stride?: number | number[],
    padding?: number | number[],
    dilation?: number | number[],
    groups?: number,
    bias?: boolean,
): Conv2d
Parameters
- in_channels: number
- out_channels: number
- kernel_size: number | number[]
- stride: number | number[] = 1
- padding: number | number[] = 0
- dilation: number | number[] = 1
- groups: number = 1
- bias: boolean = true
Returns Conv2d
Overrides _ConvNd.constructor
- Defined in nn/module.ts:142

Properties

bias

bias: Parameter

dilation

dilation: number | number[]

groups

groups: number

in_channels

in_channels: number

kernel_size

kernel_size: number | number[]

out_channels

out_channels: number

padding

padding: number | number[]

stride

stride: number | number[]

weight

weight: Parameter

Methods

forward

forward(input: Tensor): Tensor
Parameters
- input: Tensor
Returns Tensor
Overrides _ConvNd.forward
- Defined in nn/module.ts:155

parameters

parameters(): Parameter[]
Returns Parameter[]
Inherited from _ConvNd.parameters
- Defined in nn/base.ts:51

`Protected`register

register(name: string, value: Parameter | Module): void
Parameters
- name: string
- value: Parameter | Module
Returns void
Inherited from _ConvNd.register
- Defined in nn/base.ts:41