Nansum
Nansum
torch_nansum(self, dim, keepdim = FALSE, dtype = NULL)
self |
(Tensor) the input tensor. |
dim |
(int or tuple of ints) the dimension or dimensions to reduce. |
keepdim |
(bool) whether the output tensor has |
dtype |
the desired data type of returned tensor. If specified, the
input tensor is casted to dtype before the operation is performed. This is
useful for preventing data type overflows. Default: |
Returns the sum of all elements, treating Not a Numbers (NaNs) as zero.
Returns the sum of each row of the input
tensor in the given
dimension dim
, treating Not a Numbers (NaNs) as zero.
If dim
is a list of dimensions, reduce over all of them.
If keepdim
is TRUE
, the output tensor is of the same size
as input
except in the dimension(s) dim
where it is of size 1.
Otherwise, dim
is squeezed (see torch_squeeze
), resulting in the
output tensor having 1 (or len(dim)
) fewer dimension(s).
if (torch_is_installed()) { a <- torch_tensor(c(1., 2., NaN, 4.)) torch_nansum(a) torch_nansum(torch_tensor(c(1., NaN))) a <- torch_tensor(rbind(c(1, 2), c(3., NaN))) torch_nansum(a) torch_nansum(a, dim=1) torch_nansum(a, dim=2) }
Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.