tensorplay.masked

Classes

tensorplay.masked.MaskedTensor

A pair of plain tensors data and mask presented as a single value.

Functions

`tensorplay.masked.amax`	Returns maximum of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.amin`	Returns minimum of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.argmax`	Returns argmax of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.argmin`	Returns argmin of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.as_masked_tensor`
`tensorplay.masked.cumprod`	Returns cumulative_prod of all the slices in the `input` tensor along `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.cumsum`	Returns cumulative_sum of all the slices in the `input` tensor along `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.is_masked_tensor`	Return True if the input is a MaskedTensor, else False.
`tensorplay.masked.log_softmax`	Returns log_softmax of all the slices in the `input` tensor along `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.logaddexp`	Returns logaddexp of all the elements in the `input` and the `other` tensor.
`tensorplay.masked.logsumexp`	Returns logsumexp of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.masked_tensor`
`tensorplay.masked.mean`	Returns mean of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.median`	Returns median of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.norm`	Returns norm of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.normalize`	Returns normalize of all the slices in the `input` tensor along `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.prod`	Returns product of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.softmax`	Returns softmax of all the slices in the `input` tensor along `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.softmin`	Returns softmin of all the slices in the `input` tensor along `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.std`	Returns standard_deviation of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.sum`	Returns sum of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.
`tensorplay.masked.var`	Returns variance of all the elements in the `input` tensor along the given dimension(s) `dim` while the `input` elements are masked out according to the boolean tensor `mask`.

tensorplay.masked

Classes

Functions

Help improve this page

On this page