Expand description
Common dependent type patterns.
Functionsยง
- attention_
tensor - Attention tensor (batch, heads, seq_len, head_dim).
- batch_
matrix - Batch of matrices.
- batch_
vector - Batch of vectors.
- identity_
matrix - Identity matrix type (square with ones on diagonal).
- image_
tensor - 4D image tensor (batch, channels, height, width).
- square_
matrix - Square matrix type.