Skip to content

Instantly share code, notes, and snippets.

@wookim3

wookim3/libtorch.py

Created October 7, 2020 15:36
Show Gist options
  • Save wookim3/fde2d08da0e39e7e34a28a7ac350b5ec to your computer and use it in GitHub Desktop.
Save wookim3/fde2d08da0e39e7e34a28a7ac350b5ec to your computer and use it in GitHub Desktop.
Download ZIP
Raw
libtorch.py
# Python
output = torch.where(score_over_threshold, label, unknown_labels)
// C++
const auto output = torch::where(score_over_threshold, label, unknown_labels);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment