Function InplaceView

oneflow/core/framework/tensor_methods.cpp:121–135

}

Maybe<void> InplaceView(const std::shared_ptr<Tensor>& input, const Shape& target_shape,
                        const Stride& target_stride, const int64_t storage_offset) {
  Symbol<LocalTensorMeta> new_tensor_meta =
      SymbolOf(LocalTensorMeta(target_shape, target_stride, input->dtype()->data_type(),
                               input->memory_format(), JUST(input->device())));

  bool requires_grad = (autograd::GradMode::is_enabled() && input->requires_grad());
  std::shared_ptr<EagerLocalTensorImpl> new_tensor_impl = std::make_shared<EagerLocalTensorImpl>(
      JUST(input->tensor_storage()), storage_offset, /*requires_grad=*/requires_grad,
      /*is_leaf=*/!requires_grad);
  JUST(new_tensor_impl->InitEagerBlobObject(
      new_tensor_meta, JUST(JUST(input->eager_blob_object())->compute_local_dep_object())));
  JUST(JUST(input->AsLocalTensor())->set_impl(new_tensor_impl));
  return Maybe<void>::Ok();
}

Maybe<Tensor> Reshape(const std::shared_ptr<Tensor>& input, const Shape& target_shape) {
  Stride target_stride(target_shape);