aten/src/ATen/native/EmbeddingBag.cpp
// embedding_bag wrapper to enforce contiguity in tensors other than `weight`.
// This is created to save extra `.contiguous()` call in backward.
// See NOTE [ embedding_bag Native Functions ] in native_functions.yaml for details
embedding_bag wrapper to enforce contiguity in tensors other than `weight`.
This is created to save extra `.contiguous()` call in backward.
See NOTE [ embedding_bag Native Functions ] in native_functions.yaml for details
embedding_bag wrapper to enforce contiguity in tensors other than ` weight ` . This is created to save extra ` . contiguous () ` call in backward . See NOTE [ embedding_bag Native Functions ] in native_functions . yaml for details
std::tuple<Tensor, Tensor, Tensor, Tensor>
embedding_bag(const Tensor &weight, const Tensor &indices,
              const Tensor &offsets, const bool scale_grad_by_freq,
              const int64_t mode, bool sparse) {
  return at::_embedding_bag(weight, indices.contiguous(), offsets.contiguous(),
                            scale_grad_by_freq, mode, sparse);
  }
std :: tuple < Tensor , Tensor , Tensor , Tensor > embedding_bag ( const Tensor & weight , const Tensor & indices , const Tensor & offsets , const bool scale_grad_by_freq , const int64_t mode , bool sparse ) { return at :: _embedding_bag ( weight , indices . contiguous ( ) , offsets . contiguous ( ) , scale_grad_by_freq , mode , sparse ) ; }
embedding_bag wrapper to enforce contiguity in tensors other than `weight`.
embedding_bag wrapper to enforce contiguity in tensors other than ` weight ` .