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.


mlr-org/mlr3torch documentation built on April 17, 2025, 8:22 p.m.