com.intel.analytics.bigdl.nn

ops

package ops

Visibility
  1. Public
  2. All

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]

Value Members

  1. object All extends Serializable

  2. object Any extends Serializable

  3. object ApproximateEqual extends Serializable

  4. object ArgMax extends Serializable

  5. object BatchMatMul extends Serializable

  6. object BucketizedCol extends Serializable

  7. object Cast extends Serializable

  8. object CategoricalColHashBucket extends Serializable

  9. object CategoricalColVocaList extends Serializable

  10. object Ceil extends Serializable

  11. object CrossCol extends Serializable

  12. object CrossEntropy extends Serializable

  13. object DepthwiseConv2D extends Serializable

  14. object Digamma extends Serializable

  15. object Dilation2D extends Serializable

  16. object Equal extends Serializable

  17. object Erf extends Serializable

  18. object Erfc extends Serializable

  19. object Exp extends Serializable

  20. object Expm1 extends Serializable

  21. object Floor extends Serializable

  22. object FloorDiv extends Serializable

  23. object FloorMod extends Serializable

  24. object Gather extends Serializable

  25. object Greater extends Serializable

  26. object GreaterEqual extends Serializable

  27. object InTopK extends Serializable

  28. object IndicatorCol extends Serializable

  29. object Inv extends Serializable

  30. object IsFinite extends Serializable

  31. object IsInf extends Serializable

  32. object IsNan extends Serializable

  33. object Kv2Tensor extends Serializable

  34. object L2Loss extends Serializable

  35. object Less extends Serializable

  36. object LessEqual extends Serializable

  37. object Lgamma extends Serializable

  38. object LogicalAnd extends Serializable

  39. object LogicalNot extends Serializable

  40. object LogicalOr extends Serializable

  41. object Max extends Serializable

  42. object Maximum extends Serializable

  43. object Minimum extends Serializable

  44. object MkString extends Serializable

  45. object Mod extends Serializable

  46. object ModuleToOperation extends Serializable

  47. object NotEqual extends Serializable

  48. object OneHot extends Serializable

  49. object Pad extends Serializable

  50. object Pow extends Serializable

  51. object Prod extends Serializable

  52. object RandomUniform extends ModuleSerializable with Serializable

  53. object RangeOps extends Serializable

  54. object Rank extends Serializable

  55. object ResizeBilinearOps extends Serializable

  56. object Rint extends Serializable

  57. object Round extends Serializable

  58. object SegmentSum extends Serializable

  59. object Select extends Serializable

  60. object SelectTensor extends Serializable

  61. object Sign extends Serializable

  62. object Slice extends Serializable

  63. object SquaredDifference extends Serializable

  64. object Substr extends Serializable

  65. object Sum extends Serializable

  66. object TensorOp extends Serializable

  67. object Tile extends Serializable

  68. object TopK extends Serializable

  69. object TruncateDiv extends Serializable

  70. object TruncatedNormal extends Serializable

Ungrouped