contrib_sort_vertices: Contrib sort vertices

Description Usage Arguments Details Note Examples

View source: R/contrib.R

Description

Based on the implementation from Rotated_IoU

Usage

1
contrib_sort_vertices(vertices, mask, num_valid)

Arguments

vertices

A Tensor with the vertices.

mask

A tensors containing the masks.

num_valid

A integer tensors.

Details

All tensors should be on a CUDA device so this function can be used.

Note

This function does not make part of the official torch API.

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
if (torch_is_installed()) {
if (cuda_is_available()) {
v <- torch_randn(8, 1024, 24, 2)$cuda()
mean <- torch_mean(v, dim=2, keepdim=TRUE)
v <- v - mean
m <- (torch_rand(8, 1024, 24) > 0.8)$cuda()
nv <- torch_sum(m$to(dtype = torch_int()), dim=-1)$to(dtype = torch_int())$cuda()
result <- contrib_sort_vertices(v, m, nv)
}
}

torch documentation built on Oct. 7, 2021, 9:22 a.m.