Causal

Module: generative_models.core.layers.causal

Source: generative_models/core/layers/causal.py

Overview

Causal utilities for autoregressive models.

This module provides utility functions commonly used in autoregressive models, particularly for creating causal masks and shifting sequences.

Functions

apply_causal_mask

def apply_causal_mask()

create_attention_mask

def create_attention_mask()

create_causal_mask

def create_causal_mask()

shift_right

def shift_right()

Module Statistics

• Classes: 0
• Functions: 4
• Imports: 2