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 April 4, 2025, 3:03 a.m.