Известные прямые подклассы |
Нормализует TensorBuffer
с заданным средним значением и стандартным отклонением: вывод = (входное — среднее)/стандартное отклонение.
Публичные конструкторы
NormalizeOp (среднее значение с плавающей запятой, стандартное отклонение с плавающей запятой) Инициализирует NormalizeOp. | |
NormalizeOp (среднее значение float[], float[] stddev) Инициализирует NormalizeOp. |
Публичные методы
ТензорБуфер | применить (вход TensorBuffer ) Применяет определенную нормализацию к заданному тензору и возвращает результат. |
Унаследованные методы
Публичные конструкторы
public NormalizeOp (среднее значение с плавающей запятой, стандартное отклонение с плавающей запятой)
Инициализирует NormalizeOp. При вызове он создает новый TensorBuffer
, который удовлетворяет следующим требованиям:
output = (input - mean) / stddev
В следующих двух случаях сбросьте mean
на 0, а stddev
на 1, чтобы обойти нормализацию.
1. И среднее mean
, и {code stddev} равны 0.
2. mean
равно 0, а {stddev} — бесконечность.
Примечание. Если для mean
установлено значение 0, а для stddev
установлено значение 1, никаких вычислений не произойдет, и исходный ввод будет возвращен непосредственно при выполнении.
Примечание. В настоящее время возвращаемый TensorBuffer
всегда является тензором DataType.FLOAT32
, за исключением случаев, когда входными данными является тензор DataType.UINT8
, mean
установлено значение 0, а stddev
установлено значение 1, так что возвращается исходный тензор DataType.UINT8
.
Параметры
иметь в виду | среднее значение, которое необходимо вычесть первым. |
---|---|
стандартное отклонение | значение стандартного отклонения, которое затем нужно разделить. |
Броски
IllegalArgumentException | если stddev равно нулю. |
---|
public NormalizeOp (float[] среднее, float[] stddev)
Инициализирует NormalizeOp. При вызове он создает новый TensorBuffer
, который удовлетворяет следующим требованиям:
// Pseudo code. [...][i] means a certain element whose channel id is i. output[...][i] = (input[...][i] - mean[i]) / stddev[i]
Примечание. Если для всех значений mean
установлено значение 0, а для всех stddev
установлено значение 1, вычислений не произойдет, и исходный ввод будет возвращен непосредственно при выполнении.
Примечание. Возвращаемый TensorBuffer
в настоящее время всегда является тензором DataType.FLOAT32
, за исключением того, что входными данными является тензор DataType.UINT8
, все mean
установлены на 0, а все stddev
установлены на 1.
Параметры
иметь в виду | средние значения, которые необходимо вычесть в первую очередь для каждого канала. |
---|---|
стандартное отклонение | значения стандартного отклонения, которые затем нужно разделить для каждого канала. |
Броски
IllegalArgumentException | если какое-либо stddev равно нулю, или mean число элементов отличается от stddev , или любой из них пуст. |
---|
Публичные методы
общедоступное применение TensorBuffer (вход TensorBuffer )
Применяет определенную нормализацию к заданному тензору и возвращает результат.
Примечание. input
, возможно, являются тем же экземпляром, что и выходные данные.
Параметры
вход | входной тензор. Это может быть тот же экземпляр с выходом. |
---|
Возврат
- выходной тензор.