nn_reshape {mlr3torch}R Documentation

Reshape

Description

Reshape a tensor to the given shape.

Usage

nn_reshape(shape)

Arguments

shape

(integer())
The desired output shape.

[Package mlr3torch version 0.1.0 Index]