Returns a tensor filled with the scalar value 0, with the shape defined by the variable argument sizes.

tch_zeros(sizes, dtype = NULL, layout = NULL, device = NULL,
  requires_grad = FALSE)

Arguments

sizes

a sequence of integers defining the shape of the output tensor.

dtype

the desired data type of returned tensor. Default: if NULL, infers data type from x.

device

the desired device of returned tensor. Default: if NULL, uses the current device for the default tensor type (see tch_set_default_tensor_type()). device will be the CPU for CPU tensor types and the current CUDA device for CUDA tensor types.

requires_grad

If autograd should record operations on the returned tensor. Default: FALSE.

the

desired layout of returned Tensor. Default: 'strided'

Examples

tch_zeros(c(2, 4))
#> tensor #> 0 0 0 0 #> 0 0 0 0 #> [ Variable[CPUFloatType]{2,4} ]
tch_zeros(5)
#> tensor #> 0 #> 0 #> 0 #> 0 #> 0 #> [ Variable[CPUFloatType]{5} ]