Marian NMT
v1.11.5
Contents:
Code Organisation
Expression graphs
Operations in the expression graph
Layers
Using Marian with factors
Library API
How to contribute to Marian
Writing documentation
Marian NMT
»
Library API
»
Function marian::gpu::SanitizeGradient
View page source
Function marian::gpu::SanitizeGradient
¶
Defined in
File tensor_operators.cu
Function Documentation
¶
bool
marian::
gpu
::
SanitizeGradient
(
marian::Tensor
in
,
Ptr
<
Allocator
>
allocator
, bool
pruneNaN
, bool
clipInf
)
¶