-
public class Operator
-
-
Method Summary
Modifier and Type Method Description final static Unitaddmv(MTensor x, MTensor b)final static MTensormul(MTensor x, MTensor w)final static Unitrelu(MTensor x)final static Unitflatten(MTensor x, Integer startDim)final static MTensorconcatenate(Array<MTensor> tensors)final static Unitsoftmax(MTensor x)final static MTensordense(MTensor x, MTensor w, MTensor b)final static MTensorembedding(Array<String> texts, Integer seqLength, MTensor w)final static MTensortranspose2D(MTensor x)final static MTensortranspose3D(MTensor x)final static MTensorconv1D(MTensor x, MTensor w)final static MTensormaxPool1D(MTensor x, Integer poolSize)-
-
Method Detail
-
concatenate
final static MTensor concatenate(Array<MTensor> tensors)
-
transpose2D
final static MTensor transpose2D(MTensor x)
-
transpose3D
final static MTensor transpose3D(MTensor x)
-
-
-
-