Method batch_elem

dynet/tensor.h:92–106

* \brief Get a Tensor object representing a single batch. * \details If this tensor only has a single batch, then broadcast. Otherwise, check to make sure that the requested batch is smaller than the number of batches. * * TODO: This is a bit wasteful, as it re-calculates `bs.batch_size()` every time. * * \param b Batch id * \return Sub tensor at batch `b` */

   * \return Sub tensor at batch `b`
   */
  Tensor batch_elem(unsigned b) const {
    if (d.batch_elems() == 1) {
      return *this;
    } else {
      if (b >= d.batch_elems()) {
        std::stringstream ss;
        ss << "Requested batch id " << b << " is greater than the number of batch " << d.batch_elems();
        throw std::runtime_error(ss.str());
      }
      const unsigned bsize = d.batch_size();
      Dim new_d(d); new_d.bd = 1;
      Tensor ret(new_d, v + bsize * b, device, mem_pool);
      return ret;
    }
  }

  // get tensors for all batches
  /**