Skip to content

Commit 6c62cca

Browse files
committed
Added dtype to normalize jsdoc
1 parent fe38226 commit 6c62cca

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/torchvision/transforms/transforms.js

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ export class InvertAxes extends Function {
2929
export class Normalize extends Function {
3030
/**
3131
* Create a new callable Normalize object
32-
* @param {*} mean A set of means across dimensions
33-
* @param {*} std A set of standard deviation values across dimensions
32+
* @param {Array<number>} mean A set of means across dimensions
33+
* @param {Array<number>} std A set of standard deviation values across dimensions
3434
*/
3535
constructor(mean, std) {
3636
super();

0 commit comments

Comments
 (0)