Crate caffe2op_batch

source ·

Macros

Structs

  • | Input data is a N * D matrix. Apply box-cox | transform for each column. lambda1 and lambda2 | is of size D that defines the hyper-parameters for | the transform of each column x of the input | data: | | ln(x + lambda2), if lambda1 == 0 | ((x + lambda2)^lambda1 - 1)/lambda1, if lambda1 != 0 |
  • | Bucketize the float_features into | sparse features. | | The float_features is a N * D tensor where | N is the batch_size, and D is the feature_dim. | | The indices is a 1D tensor containing | the indices of the features that need | to be bucketized. | | The lengths is a 1D tensor that splits | the following ‘boundaries’ argument. | | The boundaries is a 1D tensor containing | the border list for each feature. | | With in each batch, indices should | not have duplicate number, and the number | of elements in indices should be less | than or equal to D. | | Each element in lengths vector (lengths[i]) | represents the number of boundaries | in the sub border list. | | The sum of all elements in lengths | must be equal to the size of boundaries. | | If lengths[0] = 2, the first sub border | list is [0.5, 1.0], which separate the | value to (-inf, 0.5], (0,5, 1.0], (1.0, | inf). The bucketized feature will have | three possible values (i.e. 0, 1, 2). |
  • | This Op is a inverse of BatchSparseToDenseOp. | | Basically, given a lengths vector, a indices | vector, and a dense matrix dense, output value | vector so that, along with lengths vector and | indices vector, forms a sparse representation of | the dense matrix. | | A sparse matrix is represented by lengths | vector, indices vector, and values vector. | | Each element in lengths vector (lengths[i]) | represents the number of indices in this batch | (batch i). | | With in each batch, indices should not have | duplicate number. |
  • | Batch gather operation, first dimension in DATA is | the batch size. | | Given DATA tensor of rank r >= 2, and INDICES | tensor of rank q >= 1, gather entries of the | second outer dimension (axis == 1) of DATA indexed | by INDICES, and concatenate them in an output | tensor of rank q + (r - 1).
  • | Batch Matrix multiplication Yi = Ai * Bi, where | A has shape (dim0, dim1, … M, K), B has shape | (dim0, dim1, … K, N), Y has shape (dim0, dim1, | … M, N) and i ranges from 0 to (dim0 * dim1 …) | - 1. rank(A) == rank(B) >= 2. | | In case of A and B being two dimensional, it | behaves like normal matrix multiplication.
  • | Batch permutation of an input tensor | X given input indices. | | First dimension of X equals batch size | N. | | The indices stores a be permutation | of N. | | The output Y is a tensor of same shape | as X, with data re-ordered according | to the indices within the batch size. |
  • | Convert sparse matrix representation | into dense matrix. | | A sparse matrix is represented by lengths | vector, indices vector, and values | vector. | | Each element in lengths vector (lengths[i]) | represents the number of indices in | this batch (batch i). | | With in each batch, indices should | not have duplicate number. |
  • | Buffers used by the MKL version are cached | across calls. |

Functions

Type Definitions