Module ml_ops

Module ml_ops 

Source
Expand description

Machine learning operations using the array protocol.

This module provides implementations of various machine learning operations using the array protocol, such as activation functions, convolution, and pooling.

Enums§

ActivationFunc
Activation function types.

Functions§

activation
batch_norm
conv2d
cross_entropy
dropout
max_pool2d
self_attention