ConvolutionAxes

import { ConvolutionAxes } from "destack:tensor";

Axis mapping for tensor convolution.

export struct ConvolutionAxes<const SpatialRank: usize> {
    inputBatch: Axis;
    inputFeature: Axis;
    inputSpatial: [Axis; SpatialRank];
    kernelInputFeature: Axis;
    kernelOutputFeature: Axis;
    kernelSpatial: [Axis; SpatialRank];
    outputBatch: Axis;
    outputFeature: Axis;
    outputSpatial: [Axis; SpatialRank];
}
language/library/src/tensor/axes.ds:21:48

Members

inputBatch: Axis

Batch axis in the input tensor.

inputFeature: Axis

Feature axis in the input tensor.

inputSpatial: [Axis; SpatialRank]

Spatial axes in the input tensor.

kernelInputFeature: Axis

Input feature axis in the kernel tensor.

kernelOutputFeature: Axis

Output feature axis in the kernel tensor.

kernelSpatial: [Axis; SpatialRank]

Spatial axes in the kernel tensor.

outputBatch: Axis

Batch axis in the output tensor.

outputFeature: Axis

Feature axis in the output tensor.

outputSpatial: [Axis; SpatialRank]

Spatial axes in the output tensor.