22HLT01 QUMPHY Software Package
Install
Usage
Writing a Config File
App package
QUMPHY package
Subpackages
qumphy.callbacks package
qumphy.data package
qumphy.evaluate package
qumphy.misc package
qumphy.models package
Subpackages
Submodules
Submodules
22HLT01 QUMPHY Software Package
QUMPHY package
qumphy.models package
qumphy.models.utils package
qumphy.models.utils.masking module
View page source
qumphy.models.utils.masking module
class
qumphy.models.utils.masking.
ProbMask
(
B
,
H
,
L
,
index
,
scores
,
device
=
'cpu'
)
[source]
Bases:
object
property
mask
class
qumphy.models.utils.masking.
TriangularCausalMask
(
B
,
L
,
device
=
'cpu'
)
[source]
Bases:
object
property
mask