13 lines
304 B
Python
13 lines
304 B
Python
"""Causality axis enum for specifying causal convolution dimensions."""
|
|
|
|
from enum import Enum
|
|
|
|
|
|
class CausalityAxis(Enum):
|
|
"""Enum for specifying the causality axis in causal convolutions."""
|
|
|
|
NONE = None
|
|
WIDTH = "width"
|
|
HEIGHT = "height"
|
|
WIDTH_COMPATIBILITY = "width-compatibility"
|