ops

Type Members

1. class Assert[T] extends Operation[Table, Activity, T]

   Assert will assert the first input to be true, if not, throw the message in the second input.

2. class Assign[T] extends Operation[Table, Tensor[_], T]

   Update 'ref' by assigning 'value' to it.

3. class AssignGrad[T] extends Operation[Tensor[T], Activity, T]

4. class BiasAddGrad[T] extends Operation[Tensor[T], Tensor[T], T]

5. class BroadcastGradientArgs[T] extends Operation[Table, Table, T]

   Given shapes of two tensors, computes the reduction indices for the gradient computation.

6. class Cast[T, D] extends Operation[Tensor[_], Tensor[D], T]

   Casts a tensor to a new type.

7. sealed abstract class ControlOps[T] extends Operation[Activity, Activity, T]

   Control flow related operations

8. class Conv2D[T] extends Operation[Table, Tensor[T], T]

9. class Conv2DBackFilter[T] extends Operation[Activity, Tensor[T], T]

10. class Conv2DTranspose[T] extends Operation[Activity, Tensor[T], T]

    Backward of SpatialConvolution

11. class CrossEntropy[T] extends Operation[Table, Table, T]

    Compute the cross entropy loss and the gradients.

12. class DecodeGif[T] extends DecodeImage[T]

13. class DecodeImage[T] extends Operation[Tensor[ByteString], Tensor[Int], T]

14. class DecodeJpeg[T] extends DecodeImage[T]

15. class DecodePng[T] extends DecodeImage[T]

16. class DecodeRaw[T] extends Operation[Tensor[ByteString], Activity, T]

17. class Equal[T] extends Operation[Table, Tensor[Boolean], T]

18. class Floor[T] extends Operation[Tensor[_], Tensor[_], T]

19. class Greater[T] extends Operation[Table, Tensor[Boolean], T]

20. abstract class IdentityControl[T] extends ControlOps[T]

    Control flow related operations, and they just pass the input without modifying them

21. class L2Loss[T] extends Operation[Tensor[_], Tensor[_], T]

22. class Less[T] extends Operation[Table, Tensor[Boolean], T]

23. class LogicalAnd[T] extends Operation[Table, Tensor[Boolean], T]

24. class LogicalNot[T] extends Operation[Tensor[Boolean], Tensor[Boolean], T]

25. class LogicalOr[T] extends Operation[Table, Tensor[Boolean], T]

26. class MaxPool[T] extends Operation[Tensor[T], Tensor[T], T]

27. class MaxPoolGrad[T] extends Operation[Table, Tensor[T], T]

28. sealed class MergeControlNode[T] extends Node[T]

    A wrapper of node for merge operation.

29. class ModuleToOperation[T] extends Operation[Activity, Activity, T]

    Wrap a nn module to an Operation

30. class NoOp[T] extends Operation[Activity, Activity, T] with WithoutInput

31. class NotEqual[T] extends Operation[Table, Tensor[Boolean], T]

32. class OneHot[T, D] extends Operation[Table, Tensor[D], T]

    OneHot operation returns a one-hot tensor

33. abstract class Operation[A <: Activity, B <: Activity, T] extends AbstractModule[A, B, T]

    Operation is an abstract class which represents the most basic operations An operations has only forward functions and without backward functions.

34. class Pad[T, D] extends Operation[Table, Tensor[D], T]

35. class ParseExample[T] extends Operation[Table, Table, T]

36. class Pow[T] extends Operation[Table, Tensor[_], T]

37. class Prod[T] extends Operation[Tensor[_], Tensor[_], T]

38. class RandomUniform[T, D] extends Operation[Tensor[Int], Tensor[D], T] with RandomNode

39. class Rank[T] extends Operation[Tensor[_], Tensor[Int], T]

40. class ReluGrad[T] extends Operation[Table, Tensor[T], T]

41. class ResizeBilinearOps[T] extends Operation[Activity, Tensor[Float], T]

42. class Select[T] extends Operation[Table, Activity, T]

    Selects elements from input, depending on given condition.

43. class Slice[T] extends Operation[Tensor[_], Tensor[_], T]

    This operation extracts a slice of size size from a tensor input starting at the location specified by begin.

44. class Substr[T] extends Operation[Table, Tensor[ByteString], T]

45. class Sum[T, D] extends Operation[Table, Tensor[D], T]

46. sealed class SwitchControlNode[T] extends Node[T]

    A wrapper of node for switch operation.

47. class Tile[T] extends Operation[Table, Tensor[_], T]

    This operation creates a new tensor by replicating input multiples times.

48. class TruncatedNormal[T, DataType] extends Operation[Tensor[Int], Tensor[DataType], T]