Method GetOutputSampleShape

include/singa/model/layer.h:84–87

Return the shape of the generated Tensor without the batchsize dimension

  /// Return the shape of the generated Tensor without the batchsize dimension
  virtual const Shape GetOutputSampleShape() const {
    LOG(FATAL) << "Pls override this function";
    return vector<size_t>{};
  }
  /// Return the shape of the k-th generated tensor without the batchsize
  /// dimension. Used for layers that generate multiple tensors.
  virtual const Shape GetOutputSampleShape(int k) {