package kaun

  1. Overview
  2. Docs
package kaun
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Kaun.AttentionSource

Sourcemodule Ptree = Ptree
Sourceval normalize_mask : ('a, 'layout) Rune.t -> Rune.bool_t

Convert any numeric/bool mask into a boolean tensor (non-zero => true).

Sourcemodule Multi_head : sig ... end