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::Select
View page source
Function marian::gpu::Select
¶
Defined in
File tensor_operators.cu
Function Documentation
¶
void
marian::
gpu
::
Select
(
Tensor
out
,
const
Tensor
in
,
const
Tensor
indices
, int
axis
)
¶