nn_reshape: Reshape

nn_reshapeR Documentation

Reshape

Description

Reshape a tensor to the given shape.

Usage

nn_reshape(shape)

Arguments

shape

(integer())
The desired output shape.


mlr3torch documentation built on Aug. 26, 2025, 5:09 p.m.