Why Gemfury?
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Debian packages
RPM packages
NuGet packages
duality-group / torch python
Repository URL to install this package:
|
Version:
2.1.2+cpu ▾
|
#pragma once
#include <c10/core/ScalarType.h>
#include <cstdint>
namespace at {
struct TensorIteratorBase;
class TensorBase;
}
namespace at {
namespace native {
/// @param maskPrefixSum[in,out]
void launch_masked_scatter_kernel(
const TensorBase &self, const TensorBase &mask,
const TensorBase &maskPrefixSum, const TensorBase &source);
}}