Skip to contents

Indicates that a tensor is a buffer in a nn_module

Usage

nn_buffer(x, persistent = TRUE)

Arguments

x

the tensor that will be converted to nn_buffer

persistent

whether the buffer should be persistent or not.