Class IdentityImpl

• Defined in File linear.h

Inheritance Relationships

Base Type

• public torch::nn::Cloneable< IdentityImpl > (Template Class Cloneable)

Class Documentation

class IdentityImpl : public torch::nn::Cloneable<IdentityImpl>

A placeholder identity operator that is argument-insensitive.

See https://pytorch.org/docs/main/generated/torch.nn.Identity.html to learn about the exact behavior of this module.

Public Functions

virtual void reset() override

reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

virtual void pretty_print(std::ostream &stream) const override

Pretty prints the Identity module into the given stream.

Tensor forward(const Tensor &input)