快捷方式

Delta

class torchrl.modules.Delta(param: Tensor, atol: float = 1e-06, rtol: float = 1e-06, batch_shape: Optional[Union[Size, Sequence[int]]] = None, event_shape: Optional[Union[Size, Sequence[int]]] = None)[原始碼]

Delta 分佈。

參數:
  • param (torch.Tensor) – delta 分佈的參數;

  • atol (number, optional) – 絕對容差,用於判斷張量是否符合分佈參數;預設值為 1e-6

  • rtol (number, optional) – 相對容差,用於判斷張量是否符合分佈參數;預設值為 1e-6

  • batch_shape (torch.Size, optional) – 批次形狀;

  • event_shape (torch.Size, optional) – 結果的形狀。

log_prob(value: Tensor) Tensor[source]

傳回在 value 評估的機率密度/質量函數的對數。

參數:

value (Tensor) –

property mean: Tensor

傳回分配的平均值。

property mode: Tensor

傳回分配的眾數。

rsample(size=None) Tensor[source]

如果分配參數為批次處理,則產生一個 sample_shape 形狀的重新參數化樣本,或 sample_shape 形狀的一批重新參數化樣本。

sample(size=None) Tensor[source]

如果分配參數為批次處理,則產生一個 sample_shape 形狀的樣本,或 sample_shape 形狀的一批樣本。

文件

存取 PyTorch 的完整開發人員文件

檢視文件

教學課程

取得初學者和進階開發人員的深入教學課程

檢視教學課程

資源

尋找開發資源並獲得問題解答

檢視資源