Class Conv2d

Hierarchy

_ConvNd
- Conv2d

Index

Constructors

Properties

bias dilation groups in_channels kernel_size out_channels padding stride training weight

Methods

call eval forward named_parameters parameters register train

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:168

Properties

bias

bias: nn.parameter.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[]

training

training: boolean = true

weight

weight: nn.parameter.Parameter

Methods

call

call(...args: Tensor[]): Tensor
Entry point for running the module. Equivalent to model(x) in Python. In the future, this is where forward hooks will be triggered. Call forward() directly to bypass hooks.
Parameters
- ...args: Tensor[]
Returns Tensor
Inherited from _ConvNd.call
- Defined in nn/base.ts:39

eval

eval(): this
Returns this
Inherited from _ConvNd.eval
- Defined in nn/base.ts:51

forward

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

named_parameters

named_parameters(prefix?: string): [string, nn.parameter.Parameter][]
Parameters
- prefix: string = ''
Returns [string, nn.parameter.Parameter][]
Inherited from _ConvNd.named_parameters
- Defined in nn/base.ts:63

parameters

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

`Protected`register

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

train

train(mode?: boolean): this
Parameters
- mode: boolean = true
Returns this
Inherited from _ConvNd.train
- Defined in nn/base.ts:43