C++ (Cpp) THCSTensor_ Examples

Example #1

File: THCSTensor.cpp Project: xiongyw/pytorch

int64_t THCSTensor_(size)(THCState *state, const THCSTensor *self, int dim)
{
  THArgCheck((dim >= 0) && (dim < self->nDimensionI + self->nDimensionV),
      1, "dimension %d out of range of %dD tensor",
      dim+1, THCSTensor_(_nDimension)(state, self));
  return self->size[dim];
}

Example #2

File: THCSTensor.cpp Project: athiwatp/pytorch

THCSTensor<real>::~THCSTensor() {
  if (tensor)
    THCSTensor_(free)(state, tensor);
}

Example #3

File: THCSTensor.cpp Project: xiongyw/pytorch

THCSTensor* THCSTensor_(_set)(THCState *state, THCSTensor *self, THCIndexTensor *indices, THCTensor *values) {
  // Note: Not like torch.set, this is an internal method
  return THCSTensor_(_move)(state, self, THCIndexTensor_(newClone)(state, indices), THCTensor_(newClone)(state, values));
}