Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::autograd::InputMetadata Struct Reference

A tensor's type and shape. More...

#include <input_metadata.h>

Public Member Functions

 InputMetadata (const at::Type &type, at::IntArrayRef shape, at::Device device)
 InputMetadata (const at::Tensor &t)
bool is_valid () const
const at::Typetype () const
at::IntArrayRef shape () const
at::Device device () const
at::Tensor zeros_like () const

Detailed Description

A tensor's type and shape.

Each Function records the required type and shape of its inputs. If is_valid() is false, then the corresponding input is not used and may be an undefined tensor.

Definition at line 12 of file input_metadata.h.

The documentation for this struct was generated from the following file: