Caten Documentation

  • Home
  • Quickstart
  • Development
  • API Reference
    • caten/air
    • caten/aasm
    • caten/codegen
    • caten/api
      • Overview
      • Tensor
      • Func
      • Module
      • Model
      • Initializers
      • ShapeTracker
      • Facet API
      • StateDict
    • caten/nn
      • Activation
      • Convolution
      • Criterion
      • Embedding
      • Linear
      • Normalization
      • Padding
      • Pooling
      • Encoding
      • Optimizers
  • Ready to use packages
    • Overview
    • caten/apps.gpt2
  • External Packages
    • caten/gguf
    • caten/oonx
    • caten/llm
In this article
  • Convolutions
    • [class] ConvND
    • [function] !convnd

Convolution

  1. Caten Documentation
  2. API Reference
  3. caten/nn
  4. Convolution
|
  • Share via

  •  Edit this article

Convolutions

[class] ConvND

(ConvND in-channels out-channels kernel-size &key (groups 1) (stride 1) (dilation 1) (padding 0) (bias t))

Applies a convolution over a tensor with a given weight and optional bias. NOTE: unlike PyTorch, this implementation is not limited to only 2d convolutions and instead works for any number of dimensions.

[function] !convnd

(!convnd x weight &key (bias nil) (groups 1) (stride 1) (dilation 1) (padding 0))
Applies a convolutional layer over a tensor x with a given weight and optional bias.

Search
Enter a keyword to search.