EfficientAttentionLayer3D: Efficient attention layer (3-D)

EfficientAttentionLayer3DR Documentation

Efficient attention layer (3-D)

Description

Efficient attention layer (3-D)

Value

output of tensor shape.

Arguments

numberOfFiltersFG

number of filters for F and G layers.

numberOfFiltersH

number of filters for H. If = NA, only use filter F for efficiency.

poolSize

pool_size in max pool layer.

doConcatenateFinalLayers

concatenate final layer with input. Alternatively, add.

Details

$initialize instantiates a new class.

$call main body.

$compute_output_shape computes the output shape.

Author(s)

BB Avants, NJ Tustison


ANTsX/ANTsRNet documentation built on April 28, 2024, 12:16 p.m.