ops

Type Members

1. class All[T] extends Operation[Table, Tensor[Boolean], T]

2. class Any[T] extends Operation[Table, Tensor[Boolean], T]

3. class ApproximateEqual[T] extends Compare[T]

4. class ArgMax[T] extends Operation[Table, Tensor[Int], T]

5. class BatchMatMul[T, D] extends Operation[Table, Tensor[D], T]

   Multiplies all slices of Tensor x and y (each slice can be viewed as an element of a batch), and arranges the individual results in a single output tensor of the same batch size.

6. class BucketizedCol[T] extends Operation[Tensor[T], Tensor[Int], T]

   BucketizedCol operation represents discretized dense input.

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

   Casts a tensor to a new type.

8. class CategoricalColHashBucket[T] extends Operation[Tensor[String], Tensor[Int], T]

   CategoricalColHashBucket operation can convert feature string to a Sparse/Dense Tensor

9. class CategoricalColVocaList[T] extends Operation[Tensor[String], Tensor[Int], T]

   CategoricalColVocaList operation having an vocabulary mapping feature string to Integer ID

10. class Ceil[T, D] extends Operation[Tensor[D], Tensor[D], T]

11. abstract class Compare[T] extends Operation[Table, Tensor[Boolean], T]

12. class CrossCol[T] extends Operation[Table, Tensor[Int], T]

    CrossCol operation preforms crosses of categorical features.

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

    Compute the cross entropy loss and the gradients.

14. class DepthwiseConv2D[T] extends Operation[Table, Tensor[T], T]

15. class Digamma[T, D] extends Operation[Tensor[D], Tensor[D], T]

16. class Dilation2D[T, D] extends Operation[Table, Tensor[D], T]

    Computes the grayscale dilation of 4-D input and 3-D filter tensors.

17. class Equal[T] extends Compare[T]

18. class Erf[T, D] extends Operation[Tensor[D], Tensor[D], T]

19. class Erfc[T, D] extends Operation[Tensor[D], Tensor[D], T]

20. class Exp[T, D] extends Operation[Tensor[D], Tensor[D], T]

21. class Expm1[T, D] extends Operation[Tensor[D], Tensor[D], T]

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

23. class FloorDiv[T, D] extends Operation[Table, Tensor[D], T]

24. class FloorMod[T, D] extends Operation[Table, Tensor[D], T]

25. class Gather[T, D] extends Operation[Table, Tensor[D], T]

    Gather slices from first input tensor according to the second input tensor.

26. class Greater[T] extends Compare[T]

27. class GreaterEqual[T] extends Compare[T]

28. class InTopK[T] extends Operation[Table, Tensor[Boolean], T]

29. class IndicatorCol[T] extends Operation[Tensor[Int], Tensor[T], T]

    Indicator operation represents multi-hot representation of given Tensor.

30. class Inv[T, D] extends Operation[Tensor[D], Tensor[D], T]

31. class IsFinite[T, D] extends Operation[Tensor[D], Tensor[Boolean], T]

32. class IsInf[T, D] extends Operation[Tensor[D], Tensor[Boolean], T]

33. class IsNan[T, D] extends Operation[Tensor[D], Tensor[Boolean], T]

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

    Kv2Tensor operation convert a kv feature column to a SparseTensor or DenseTensor

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

36. class Less[T] extends Compare[T]

37. class LessEqual[T] extends Compare[T]

38. class Lgamma[T, D] extends Operation[Tensor[D], Tensor[D], T]

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

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

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

42. class Max[T, D] extends Operation[Table, Tensor[D], T]

    Computes the maximum of elements across dimensions of a tensor.

43. class Maximum[T, D] extends Operation[Table, Tensor[D], T]

44. class Minimum[T, D] extends Operation[Table, Tensor[D], T]

45. class MkString[T] extends Operation[Tensor[_], Tensor[String], T]

    MkString operation converts a SparseTensor/DenseTensor to a Dense Tensor[String]

46. class Mod[T, D] extends Operation[Table, Tensor[D], T]

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

    Wrap a nn module to an Operation

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

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

    OneHot operation returns a one-hot tensor

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

    Operation is an abstract class which represents a forward only layer.

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

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

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

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

55. class RangeOps[T, D] extends Operation[Table, Tensor[D], T]

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

57. class ResizeBilinearOps[T] extends Operation[Activity, Tensor[T], T]

58. class Rint[T] extends Operation[Tensor[Float], Tensor[Float], T]

59. class Round[T, D] extends Operation[Tensor[D], Tensor[D], T]

60. class SegmentSum[T] extends Operation[Table, Tensor[T], T]

    Computes the sum along segments of a tensor.

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

    Selects elements from input, depending on given condition.

62. class SelectTensor[T] extends Operation[Table, Tensor[T], T]

    Select and copy a Tensor from a Table with a key.

63. class Sign[T, D] extends Operation[Tensor[D], Tensor[D], T]

64. 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.

65. class SquaredDifference[T] extends Operation[Table, Tensor[_], T]

    Returns (x - y)(x - y) element-wise.

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

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

68. class TensorOp[T] extends Operation[Tensor[T], Tensor[T], T]

    TensorOp is an Operation with Tensor[T]-formatted input and output, which provides shortcuts to build Operations for tensor transformation by closures.

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

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

70. class TopK[T, D] extends Operation[Tensor[D], Table, T]

71. class TruncateDiv[T, D] extends Operation[Table, Tensor[D], T]

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