Skip to contents

Reshape

Source: R/PipeOpTorchReshape.R
nn_reshape.Rd

Reshape a tensor to the given shape.

Usage

nn_reshape(shape)

Arguments

shape

(integer())
The desired output shape.