Shortcuts

torch.nn.functional.threshold

torch.nn.functional.threshold(input, threshold, value, inplace=False)

对输入张量的每个元素应用一个阈值。

更多详情请参见 Threshold

Return type

张量

优云智算