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::Type &	type () 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:

• torch/csrc/autograd/input_metadata.h